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