V - Infinite Prefixes CodeForces - 1295B math

天哪！！菜到家啦。

数学+思维。

首先求出一个周期内cnt0-cnt1=c的个数，如果C=0，那么只要在一个周期内有前缀等于x，那么答案就是-1，否则答案就是0

如果C!=0，列一下方程x=t*c+a.即x-a=t*c。，要求t为大于等于0的整数，也就是我们的判断条件。。。。唉。

#include<bits/stdc++.h>
using namespace std;
void solve()
{
    int cnt1=,cnt0=;
    int n,x;
    cin>>n>>x;
    int ans=;
    if(x==) ans++;
    string s;
    cin>>s;
    for(int i=;i<n;i++){
        if(s[i]=='') cnt0++;
        else cnt1++;
    }
    int c=cnt0-cnt1;
    if(c==){
        if(x==){
            cout<<-<<endl;
            return ;
        }
        else {
            cnt1=;
            cnt0=;
            for(int i=;i<n;i++){
                if(s[i]=='') cnt1++;
                else cnt0++;
                if(cnt0-cnt1==x) ans++;
            }
            if(ans>) cout<<-<<endl;
            else cout<<<<endl;
        }
    }
    else {
        cnt0=;
        cnt1=;
        for(int i=;i<n;i++){
            if(s[i]=='') {
               cnt1++;
            }
            else cnt0++;
            int a=cnt0-cnt1;
            if((x-a)%c==&&(x-a)/c>=) ans++;
        }
        cout<<ans<<endl;

    }

    return ;
}

int main()
{
    int t;
    cin>>t;
    while(t--) solve();
    return ;
}