CSES - TCR Contest - Results
Submission details
Task:Bit strings
Sender:Kalle Luopajärvi
Submission time:2017-11-21 19:41:07 +0200
Language:C++
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.04 sdetails
#2ACCEPTED0.16 sdetails
#3ACCEPTED0.17 sdetails
#4ACCEPTED0.16 sdetails
#5ACCEPTED0.19 sdetails
#6ACCEPTED0.30 sdetails

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:127:29: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i = 0; i < q.size(); ++i) {
                             ^

Code

#include <bits/stdc++.h>
using namespace std;
typedef long double ld;
typedef long long ll;
typedef complex<ld> co;
const ld PI=atan2((ld)0, (ld)-1);
void fft(vector<co>&a, int n, int k) {
	vector<co> ww(n+1);
	ww[1]=co(1, 0);
	for (int t=0;t<k-1;t++) {
		co c=polar((ld)1, PI/n*(1<<(k-1-t)));
		int p2=(1<<t),p3=p2*2;
		for (int j=p2;j<p3;j++) ww[j*2+1]=(ww[j*2]=ww[j])*c;
	}
	for (int i=0;i<n;i++) {
		int u=0;
		for (int j=1;j<n;j*=2) {u*=2;if (i&j) u++;}
		if (i<u) swap(a[i], a[u]);
	}
	for (int l=1;l<n;l*=2) {
		for (int i=0;i<n;i+=l) {
			for (int it=0,j=i+l,w=l;it<l;it++,i++,j++) {
				co t=a[j]*ww[w++];
				a[j]=a[i]-t;
				a[i]=a[i]+t;
			}
		}
	}
}
vector<ll> conv(const vector<ll>& a, const vector<ll>& b) {
	int as=a.size(), bs=b.size();
	if (as*bs==0) return {};
	int k=0;
	while ((1<<k)<as+bs-1) k++;
	int n=1<<k;
	vector<co> c(n+1);
	for (int i=0;i<n;i++) {
		if (i<as) c[i]=a[i];
		if (i<bs) c[i]={c[i].real(), (ld)b[i]};
	}
	fft(c, n, k);
	c[n]=c[0];
	for (int i=0;i<=n-i;i++) {
		c[i]=conj(c[i]*c[i]-conj(c[n-i]*c[n-i]))*co(0,(ld)1/n/4);
		c[n-i]=conj(c[i]);
	}
	fft(c, n, k);
	vector<ll> r(as+bs-1);
	for (int i=0;i<as+bs-1;i++) r[i]=round(c[i].real());
	return r;
}
int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int tt;
	cin>>tt;
	for(int xx = 0; xx < tt; ++xx) {
		string s;
		cin>>s;
		int n = s.size();
		vector<ll> w(n);
		int zeroes = 0;
		for(int i = 0; i < n; ++i) {
			char q;
			//cin>>q;
			q = s[i];
			w[i] = q-'0';
			zeroes += (w[i] == 0);
		}
		/*
		int ans2[100] = {0};
		for(int i = 0; i < n; ++i) {
			for(int j = i; j < n; ++j) {
				int sum = 0;
				for(int k = i; k <= j; ++k) {
					sum += w[k];
				}
				ans2[sum]++;
			}
		}
		for(int i = 0; i <= n; ++i) cout<<ans2[i]<<' ';
		cout<<'\n';
		*/
		if(zeroes == n) {
			cout<<(ll)n*(n+1)/2<<' ';
			for(int i = 1; i <= n ;++i) {
				cout<<0<<' ';
			}
			cout<<'\n';
			continue;
		}
		int prev = -1;
		vector<ll> v;
		ll ans = 0;
		for(int i = 0; i < n; ++i) {
			if(w[i] == 1) {
				v.push_back(i-prev);
				ans+= ll(i-prev)*(i-prev-1)/2;
				prev = i;
			}
		}
		ans += ll(n-1-prev)*(n-prev)/2;
		prev = n;
		vector<ll> q;
		for(int i = n-1; i >= 0; --i) {
			if(w[i] == 1) {
				q.push_back(prev-i);
				prev = i;
			}
		}
		reverse(v.begin(), v.end());
		reverse(q.begin(), q.end());
		
		
		/*
		for(auto x: v) cout<<x<<' ';
		cout<<'\n';
		for(auto x: q) cout<<x<<' ';
		cout<<'\n';
		cout<<endl;
		*/
		auto c = conv(v, q);

		int lol = 1;
		//	cout<<(ll)zeroes*(zeroes+1)/2<<' ';
		cout<<ans<<' ';
		for(int i = 0; i < q.size(); ++i) {
			cout<<c[q.size()-1+i]<<' ';
			++lol;
		}
		for(;lol <= n; ++lol) {
			cout<<0<<' ';
		}
		cout<<'\n';

	}
}

Test details

Test 1

Verdict: ACCEPTED

input
20
00011000100111001100
00001001111111100010
10110011110101111101
01111011010010100111
...

correct output
21 33 39 30 23 20 25 7 12 0 0 ...

user output
21 33 39 30 23 20 25 7 12 0 0 ...

Test 2

Verdict: ACCEPTED

input
1
001010110001110001110000110011...

correct output
99028 198023 199224 198569 199...

user output
99028 198023 199224 198569 199...

Test 3

Verdict: ACCEPTED

input
1
111010111001001100100101011110...

correct output
99270 199585 198291 199812 198...

user output
99270 199585 198291 199812 198...

Test 4

Verdict: ACCEPTED

input
20
001001010011101100110111110010...

correct output
5022 10235 10021 9985 10019 99...

user output
5022 10235 10021 9985 10019 99...

Test 5

Verdict: ACCEPTED

input
10
110101000110110101011011101010...

correct output
9955 20073 20158 19923 20014 1...

user output
9955 20073 20158 19923 20014 1...

Test 6

Verdict: ACCEPTED

input
1
111111111111111111111111111111...

correct output
968 100966 100967 100965 10096...

user output
968 100966 100967 100965 10096...