Codeforces 1405C

1405C
好题啊。
一开始想法是用一个sum[i]数组，遇到1 sum[i]++,遇到0 sum[i]- -,用cnt[i]表示前i位有多少个？，从k开始遍历sum，sum[i]-sum[i-k]表示长度为k的串中多出多少个0或1，用绝对值表示为res。同理cnt[i]-cnt[i-k]表示该串中？数量，如果该值大于等于res，那么该区间可以用？号表示0或1来使得区间的0，1数量相等
贴下WA代码

int main() {
	int tt;cin>>tt;
	while(tt--) {
		int n,k;
		cin>>n>>k;
		char s[3*N];
		scanf("%s",s+1);
		for(int i=1;i<=n;i++) {
			cnt[i]=cnt[i-1];
			sum[i]=sum[i-1];
		 	if(s[i]=='1') sum[i]++;
		 	else if(s[i]=='0') sum[i]--;
		 	else cnt[i]++;
		}
		int ok=1,m1=0;
		for(int i=k;i<=n;i++) {
		 	int res=abs(sum[i]-sum[i-k]+m1);
		    //int res1=cnt[i]-cnt[i-k];
		 	if(res>cnt[i]-cnt[i-k]) {
		 	 	ok=0;
		 	 	break;
		 	}
		 	else {
		 	 	if((cnt[i]-cnt[i-k]-res)%2) {
		 	 		ok=0;break; 
		 	 	}
		 	 	if(sum[i]-sum[i-1]>=0) m1-=res;
		 	 	else m1+=res;
		 	 	for(int j=i;j<=n;j++) cnt[j]-=res;
		 	}
		}
		if(ok) cout<<"YES"<<'\n';
		else cout<<"NO"<<'\n';
    }	
	return 0;
}

好好的思维题给做成模拟
AC过程
分析1100这个串， 后面一个串也符合条件11001，第4位等于第0位
也就是s[i%k]==s[i]，思维本维啊
如果s[i%k]= =’?’，这种情况下一定满足，如果s[i%k]!=s[i]那么一定不满足。

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int N=100005;
int mod=1e9+7;
int main() {
	int tt;cin>>tt;
	while(tt--) {
	int n,k;
	string s;
	cin>>n>>k>>s;
	int ok=1;
	for(int i=k;i<n;i++) {
	 	if(s[i]=='?'||s[i%k]==s[i]) continue;
	 	if(s[i%k]=='?') s[i%k]=s[i];
		else { ok=0;break; }
	}
	int cnt0=0,cnt1=0;
	for(int i=0;i<k;i++) {
	 	if(s[i]=='0') cnt0++;
	 	else if(s[i]=='1') cnt1++;
	}
	if(ok&&cnt0<=k/2&&cnt1<=k/2) cout<<"YES\n";
	else cout<<"NO\n";
	}
	return 0;
}