Binary String Reconstruction

Binary String Reconstruction

题意:

考虑如下的过程。你有一个长度为n的二进制串 $w$ 还有一个整数$x$ 。你构建了一个长度为 $n$ 的二进制串 $s$ 。二进制串 $s$ 的第$i$ 个字符串的选择如下:
1.如果$w_{i-x}$存在并且等于$1$，那么$s_i$则等于$1$。
2.如果$w_{i+x}存在并且等于$1$，那么$s_i$则等于$1$。
3.如果前两种情况都不存在，那么$s_i$ 则等于$0$。

给出字符串$s$和整数$x$，构造字符串$w$。

分析:规则告诉我们如果$s_i$为$1$，那么$w_{i-x}$和$w_{i+x}$中至少一个为$1$。如果$s_i$为$0$，那么$w_{i-x}$和$w_{i+x}$都为$0$。那么我们先把$w$全都置为$1$，把要求是$0$的地方全都变成$0$。然后检测构造后的这个字符串是否满足要求。可以见代码。

#include <bits/stdc++.h>
 
using namespace std;
const int maxn = 1e5+10;
char s[maxn];
int x;
char str[maxn];
 
int main() {
    int _;
    scanf("%d", &_);
    while (_--) {
        int flag = false;
        scanf("%s%d", s, &x);
        int len = strlen(s);
        for (int i = 0; i < len; i++) {
            str[i] = '1';
        }
        str[len]='\0';
        for (int i = 0; i < len; ++i) {
            if (s[i] == '0' && i - x >= 0) str[i - x] = '0';
            if (s[i] == '0' && i + x < len) str[i + x] = '0';
        }
        for (int i = 0; i < len; ++i) {
            if (s[i] == '0')continue;
            bool flag1 = false;
            if ((i - x >= 0 && str[i - x] == '1') || (i + x < len && str[i + x] == '1')) flag1 = true;
            if (!flag1) {
                flag = true;
                break;
            }
        }
        if (flag) puts("-1"); else printf("%s\n", str);
    }
    return 0;
}