CF 1400C 题解

题目传送门

题目解析

总感觉这道题目和CF468B的思路很像。
我们发现，如果数组 $s_i=0$ 那么 $w_{i-x}=w_{i+x}=0$ 。但是如果 $s_i=1$ 那么我们就不可以确定 $w_{i-x}$ 和 $w_{i+x}$ 的值。
所以我们先进行一次处理：首先，令所有的 $w_i=1$ 。然后当 $s_i=0$ ，那么我们就让 $w_{i-x}=w_{i+x}=0$ 。
当赋玩值之后，我们还要确保这个序列是合法的，也就是说，如果 $s_i=1$ 的话，我们就要使 $w_{i-x}=0$ 或 $w_{i+x}=0$ 否则这个序列就不存在了。我们只要输出 $-1$ 即可。

代码

#include<cstdio>
#define maxn 100039
using namespace std;
inline int read(){
    
    
	char c=getchar();
	int sum=0;
	while(c<'0'||c>'9') c=getchar();
	while('0'<=c&&c<='9'){
    
    
		sum=(sum<<1)+(sum<<3)+(c^48);
		c=getchar();
	}
	return sum;
}
char c;
int T,x,n;
int a[maxn],w[maxn];
int check(){
    
    
	int flag;
	for(int i=1;i<=n;i++)
	    if(w[i]){
    
    
	    	flag=0;
	    	if(i>x)
	    	    if(a[i-x]) flag=1;
	    	if(i+x<=n)
	    	    if(a[i+x]) flag=1;
	    	if(!flag) return 0;
		}
	return 1;
}
int main(){
    
    
	T=read();
	while(T--){
    
    
		n=0;
		c=getchar();
		while(c!='0'&&c!='1') c=getchar();
		while(c=='0'||c=='1'){
    
    
			w[++n]=c^48;
			c=getchar();
		}
		x=read();
		for(int i=1;i<=n;i++) a[i]=1;
		for(int i=1;i<=n;i++)
		    if(!w[i]){
    
    
		    	if(i>x) a[i-x]=0;
		    	if(i+x<=n) a[i+x]=0;
			}
		if(check()){
    
    
			for(int i=1;i<=n;i++)
			    printf("%d",a[i]);
			printf("\n");
		}
		else printf("-1\n");
	}
	return 0;
}