CF1400-C. Binary String Reconstruction

CF1400-C. Binary String Reconstruction

题意：

对于一个二进制字符串$s$，以及一个给定的$x$，你可以通过一下操作来得到字符串$w$：

对于字符串$s$的第$i$位，

1. 如果$i-x$有意义并且$s[i-x]==1$那么$w[i]=1$ ;

2.如果$i+x$有意义并且$s[i+x]==1$，那么$w[i]=1$;

如果上面两条都不符合，那么$w[i]=0$.

现在题目给出你字符串$w$和$x$，让你找出符合要求的字符串$s$，如果不存在这样的字符串$s$那么输出$-1$。

---

思路：

对于给出的字符串$w$，如果$w[i]==1$，那么必定有$s[i-x]=0,s[i+1]=0$。

将其他非$0$的位置补为$1$。然后按照上面给出的操作再由$s$得到$w_1$，如果$w=w_1$那么$s$就是答案，否则没有答案输出$-1$。

之所以需要由$s$再得到一遍$w_1$然后用$w$和$w_1$进行比较，原因在于：我们设$w$的一个位置$p=x$，若$w[2x]=0$，那么在$w[2x]$处就会有$s[x]=0,s[3x]=0$，而如果$w[0]=1$，那么就要求$s[x]=1$，一个位置不可能有两个值，而在上面由$w$得到$s$的过程中无法检查出这个问题。

---

#include <cstdio>
#include <cstring>
#include <algorithm>

const int Maxn = 100005;

char str[Maxn];
int s[Maxn], w[Maxn], x;

void solve() {
	scanf("%s %d", str, &x);
	int len = strlen(str);
	for (int i = 0; i < len; i++) {
		w[i] = str[i] - '0';
	}
	std::fill(s, s + len, 1);
	for (int i = 0; i < len; i++) {
		if (w[i] == 0) {
			if (i - x >= 0) {
				s[i - x] = 0;
			}
			if (i + x < len) {
				s[i + x] = 0;
			}
		}
	}
	bool flag = true;
	for (int i = 0; i < len; i++) {
		int t = 0;
		if ((i - x >= 0 && s[i - x] == 1) || (i + x) < len && s[i + x] == 1) {
			t = 1;
		}
		if (w[i] != t) {
			flag = false;
			break;
		}
	}
	if (flag) {
		for (int i = 0; i < len; i++) {
			printf("%d", s[i]);
		}
		printf("\n");
	} else {
		printf("-1\n");
	}
}

int main() {
	int T;
	scanf("%d", &T);
	while (T--) {
		solve();
	}
	return 0;
}