ural1007 Code Words (字符串处理)

三种情况分别对应m < n , m = n , m > n。然后对每一位进行枚举判断即可。有几个地方要注意一下：

1. 对于n = m的情况，只能将1改为0，而不能将0改为1。

2. 数据中字符串的数目并不是n。

3. 可能原字符串就是正确答案（样例中已经包含了这种情况）。

#include <bits/stdc++.h>
using namespace std;

char s[3010], ss[3010];

bool IsOk(char *s, int n) {
int ans = 0;
for (int i = 0; s[i] != '\0'; i++) {
if (s[i] == '1') ans += i + 1;
}
return ans % (n + 1) == 0;
}

void StrAdd(char *s, int p, char c) {
for (int i = strlen(s) + 1; i > p; i--) {
s[i] = s[i - 1];
}
s[p] = c;
}

void StrDel(char *s, int p) {
int m = strlen(s);
for (int i = p; i < m; i++) {
s[i] = s[i + 1];
}
}

int main() {
int n; scanf("%d", &n);
while (scanf("%s", s) != EOF) {
int m = strlen(s);
if (m == n) {
if (IsOk(s, n)) {
puts(s);
} else {
for (int i = 0; i < m; i++) {
strcpy(ss, s);
ss[i] = '0';
if (IsOk(ss, n)) {
puts(ss);
break;
}
}
}
} else if (m < n) {
for (int i = 0; i <= m; i++) {
strcpy(ss, s);
StrAdd(ss, i, '0');
//puts(ss);
if (IsOk(ss, n)) {
puts(ss);
break;
}
ss[i] = '1';
if (IsOk(ss, n)) {
puts(ss);
break;
}
}
} else {
for (int i = 0; i < m; i++) {
strcpy(ss, s);
StrDel(ss, i);
if (IsOk(ss, n)) {
puts(ss);
break;
}
}
}
}
return 0;
}