关于平衡字符串的问题

原题链接

对于一个01串，若其中0的个数与1的个数相同，我们称它是平衡的。给出由0、1、?三种字符组成的01串。Miaoyao想知道是否存在一种方案，将每个’?'均改变为0或1中的一种，使得所有长度为k的子串都是平衡的。
分析：
注意重点：所有长度为k的子串都是平衡的。
故对于[1,k], [2,k + 1]，其交集是[2,k], 如果这两个串都是平衡的，则s[1]必然等于s[k + 1]。
于是我们就可以根据下标分成k组，每组只要确定一个元素，那么这个组都必然是这个元素，如果一个组内出现不同的，就说明不符合。
最后，要保证1与0个数相同，故要对前k个字符进行判断，只有当1和0的数量都小于等于k/2才行。

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

const int maxv=1001000;
const int mod=998244353;
const int maxNum=0x7ffffff-10;
const double eps=1e-12;
const double PI=3.1415926535;
typedef long long LL;

char str[maxv];
bool vis[2];
int main(){
    int n,k,sum0=0,sum1=0;scanf("%d %d %s",&n,&k,str+1);
    if(k&1){printf("No");return 0;}
    for(int i=1;i<=k;i++){  //分成k组
        vis[0]=vis[1]=0;
        for(int j=i;j<=n;j+=k)   //s[i]与s[k + i]
            if(str[j]=='0'||str[j]=='1') vis[str[j]-'0']=1;
        if(vis[0]&&vis[1]){printf("No");return 0;}  //同时出现1、0，不符合
        if(vis[0]) sum0++;
        if(vis[1]) sum1++;
    }
    if(sum0<=k/2&&sum1<=k/2) printf("Yes");
    else printf("No");
    return 0;
}