APC2020 D - Anything Goes to Zero

D - Anything Goes to Zero

题意 ：给一个长度为1 ≤ N ≤ 2×10⁵的二进制串，定义f [i] 的值为 以下操作的次数：
定义 ：x_i 为二进制串的第 i 位取反后代表的十进制的值， g[ x ]为 x 的二进制表示时1的个数，设 r = x_i ，每次操作：r = r % g[ r ] ，直到 r 为 0 ；

思路： 由于第一次求余后的值必定小于N，所以关键在于先求出第一次的余数r。第一次求余的模有两个，分别是p1 = sum + 1，p2 = sum - 1，其中sum为N中1的个数。快速幂同时取模得到两个初始值cnt1，cnt2。然后再遍历二进制串，当s[i] ==0 时 用cnt1 + pow(2 , N-i-1)，当s[i] ==1 时用 cnt1 - pow(2 , N-i-1)得到第一次的余数 r ；
注意特判N中1的个数为1的情况，模不能为0；

#include<bits/stdc++.h>
using namespace std;

const int N = 2e5+7;
typedef long long LL;

int n,sum;
string s;

int quick(int a,int b,int mod)	//快速幂取模
{
	LL ans=1;
	while(b){
		if(b&1) ans = ans*a%mod;
		a = (LL)a*a%mod;
		b>>=1;
	}
	return ans;
}

int fun(int n){		//求整数n二进制表示时1的个数
	int ans=0;
	while(n){
		ans++;
		n  = n &(n-1);
	} return ans;
}

int main()
{
	cin>>n>>s;
	for(auto c :s) if(c=='1') sum++;
	int cnt1 = 0 ,cnt2 = 0, p1 = sum + 1 , p2 = sum - 1;
	
	for(int i=0;i<n;i++)
		if(s[i]=='1'){
			cnt1 = (cnt1 + quick(2,n-i-1,p1))%p1;	//两个模下的初始值
			if(p2>0) cnt2 = (cnt2 + quick(2,n-i-1,p2))%p2;
		}
	int t,ans=0;
	for(int i=0;i<n;i++){
		ans=1;
		if(s[i]=='0')  t = (cnt1+quick(2,n-i-1,p1)+p1)%p1; //得到第一次的余数 r
		else if(p2<=0) {cout<<0<<"\n"; continue;}
		else t = (cnt2-quick(2,n-i-1,p2)+p2)%p2;
		if(!t) cout<<ans<<"\n";
		else {
			while(t) ans++,t = t%fun(t);  //直接循环求次数；
			cout<<ans<<endl;
		}
	}
	return 0;
}