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;
}