题意:判断长度为n的串中的每个长度为k的子串是否0和1相等,字符串中的问号既可以是1也可以是0。
解题思路:
例如n=10 ,k=4 对于 s[0~3] 和 s[1~4] 我们可以发现1~3的部分是相同的,那么不难发现0和4相同,同理可得s[2]和s[5]也是相同的因此若该串合理就一定是k循环节子串,最后我们只需要判断前k个字母是否可以构造出一半1和一半0 ,如果1或者0的个数已经大于一半了一定构造不出来,否则一定可以构造出来。
代码:
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=300010;
bool f[N];
// x _ _ _ _ _ _ _ _ x
string s;
int main()
{
// ?00 000 100
int _;
cin>>_;
while(_--)
{
int n,k;
cin >> n >> k;
cin>>s;
bool flag=true;
for(int i=0;i<n;i++)
{
if(s[i]=='?' && s[i%k]=='?')
continue;
if(s[i]=='?' && s[i%k] != '?')
continue;
if(s[i]!='?' && s[i%k] == '?')
{
s[i%k] =s[i];
continue;
}
if(s[i]!='?' && s[i%k]!='?' && s[i]!=s[i%k])
{
flag=false;
break;
}
}
int s1=0;
int s2=0;
if(!flag)
{
cout<<"NO"<<endl;
continue;
}
for(int i=0;i<k;i++)
{
s1+=(s[i]=='0');
s2+=(s[i]=='1');
}
if(max(s1,s2)>k/2) cout<<"NO"<<endl;
else cout<<"YES"<<endl;
}
}