codeforces 1295B Infinite Prefixes 思维

https://vjudge.net/problem/CodeForces-1295B
题目大意：给定一个长度为$n$的$01$串，设其为$s$，那么字符串$t=ssss\dots \dots$，设一个字符串$q$的平衡值为$cn{t}_0-cn{t}_1$，其中$cn{t}_0$为字符串$q$中$0$的个数，$cn{t}_1$为字符串$q$中$1$的个数，现在给定字符串$s$和一个值$x$，问字符串$t$的前缀中平衡值等于$x$的前缀个数，如果这样的前缀有无穷多个，那么输出$-1$。
思路：设字符串$s$的前缀$s_i$的平衡值为$v_i$，那么字符串$t$的前缀的平衡值为：
$$v_1,v_2,v_3,\dots \dots ,v_n$$
$$v_n+v_1,v_n+v_2,v_n+v_3,\dots \dots ,2\ast v_n$$
$$2\ast v_n+v_1,2\ast v_n+v_2,2\ast v_n+v3,\dots \dots ,3\ast v_n$$
不难发现，如果平衡值等于$x$的前缀个数有无穷多个，那么必定有$v_n=0$，否则解一定是有限的。对于前缀$s_i$，如果满足$(x-v_i)\%v_n=0$且$x-v_i$和$v_n$同号，那么$s_i$就对答案有贡献。当然还有一种特殊情况我们没有考虑到，就是前缀为空字符串的时候，当且仅当$x=0$时空字符串才对答案有贡献。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=1e5+5;

char s[maxn];
int v[maxn];

int main()
{
    int t,n,x;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%s",&n,&x,s+1);
        int len=strlen(s+1);
        int cnt1=0,cnt2=0;
        bool flag=0;
        for(int i=1;i<=len;i++)
        {
            if(s[i]=='1')
                ++cnt1;
            else
                ++cnt2;
            v[i]=cnt2-cnt1;
            if(v[i]==x)
                flag=1;
        }
        if(x==0)
            flag=1;
        if(v[n]==0)
        {
            if(flag)//平衡值等于x的前缀存在 则有无穷多个
                printf("-1\n");
            else //前缀不存在 那么答案自然为0
                printf("0\n");
        }
        else
        {
            int ans=0;
            if(x==0)//前缀为空字符串时仅对x=0有贡献
                ++ans;
            for(int i=1;i<=len;i++)
                if((x-v[i])%v[n]==0&&(x>=v[i]&&v[n]>0||x<=v[i]&&v[n]<0))
                    ++ans;
            printf("%d\n",ans);
        }
    }
    return 0;
}