Educational Codeforces Round 81 (Rated for Div. 2) B - Infinite Prefixes （思维+数学）

题意：给定一个n和x，再来是一个字符串s(长度为n），现在字符串s能够无限的复制，问你有多少种可能的字符串前缀的0的个数减去1的个数能等于X
思路：模拟一下就行，不过坑点还是挺多的，设cnt为初始字符串的0和1的个数差，那么遍历初始字符串，记录每个位置的01差，只要（x-num[i])%cnt==0,那么ans++，num【i】为当前i位置的01差，不过这个还是会忽略一些情况，再把特殊情况特判一下就行。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+1;
ll num[maxn],x,cnt0,cnt1,ans;
char s[maxn];
int main()
{
	int T,n,flag;
	scanf("%d",&T);
	while(T--)
	{
		flag=cnt0=cnt1=ans=0;
		scanf("%d %lld",&n,&x);
		scanf("%s",s);
		for(int i=0;i<n;++i)
		{
			if(s[i]=='0') cnt0++;
			if(s[i]=='1') cnt1++;
			num[i+1]=cnt0-cnt1;
		}
		if(num[n]==x&&x==0)
		{
			printf("-1\n");continue;
		 } 
		 if(x==0) ans++;
		 if(num[n]==0)
		 {
		 	for(int i=1;i<=n;++i) if(num[i]==x)  {
		 		printf("-1\n");flag=1;break;
			 }
			 if(!flag) printf("0\n");
		 	continue;
		 }
		 for(int i=1;i<=n;++i)
		 {
		 	if((x-num[i])%num[n]==0&&(x-num[i])/num[n]>=0) ans++;
		 }
		 printf("%lld\n",ans);
	}
}