YC323C [ 20240724 CQYC NOIP 模擬賽 T3 ] 手環(ring）

cxqghzj發表於2024-08-19

題意

給定兩個長為 \(n\) 的 \(0/1\) 串 \(A, B\)。

每次操作：

  • 對 \(A\) 向左或向右迴圈移位。
  • 選擇 \(0 \le p < n \land B_i = 1\)，則將 \(A_i\) 取反。

求將 \(A\) 變為 \(B\) 的最小操作次數。無解輸出 -1。

\(n \le 2000\)

Sol

顯然無解當且僅當 \(A\) 和 \(B\) 不相同且 \(B\) 不存在 \(1\)。

考慮列舉一個終止狀態 \(i\)，表示 \(A\) 向右迴圈移動 \(i\) 位，且在過程中使用操作一將兩串變成相等。

考慮移動 \(i\) 位後哪些位置不滿足條件，不難發現事實上最優的決策一定滿足先向左移動，然後向右移動到 \(i\) 右邊，最後再移動回來。

原問題變為一個分配問題，每個移動 \(i\) 位後不滿足條件的位置，需要向左或是向右找到一個 \(B_i\) 為 \(1\) 的位置消掉。

因此直接預處理出每個位置向左最近的 \(B_i\) 為 \(1\) 的位置，向右最近的位置。

這樣直接對一維排序，倒著掃回來，動態維護另一維的最大值即可。

這裡直接桶排，複雜度 \(O(n ^ 2)\)。

Code

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <array>
#include <vector>
using namespace std;
#ifdef ONLINE_JUDGE

/* #define getchar() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2) ? EOF : *p1++) */
/* char buf[1 << 23], *p1 = buf, *p2 = buf, ubuf[1 << 23], *u = ubuf; */

#endif
int read() {
    int p = 0, flg = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-') flg = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        p = p * 10 + c - '0';
        c = getchar();
    }
    return p * flg;
}
void write(int x) {
    if (x < 0) {
        x = -x;
        putchar('-');
    }
    if (x > 9) {
        write(x / 10);
    }
    putchar(x % 10 + '0');
}
bool _stmer;

const int N = 2e3 + 5;

char strbuf[N];

string s, h;

array <int, N> tpl, tpr;
array <vector <int>, N> isl;

int query(int n) {
    int ans = 2 * n;
    for (int i = 0; i < n; i++) {
        int res = 0, tp = i;
        for (int j = 0; j < n; j++)
            if (s[j] != h[(j + i) % n])
                isl[tpr[j]].push_back(j), res++;
        for (int j = n; j >= i + 1; j--) {
            for (auto k : isl[j])
                tp = max(tp, tpl[(k + i) % n]);
            ans = min(ans, (tp - i + j - 1) * 2 - i + res);
            /* if ((tp - i + j - 1) * 2 - i + res < 10) { */
                /* cerr << i << " " << tp << " " << j << "#" << endl; */
                /* for (auto k : isl[j]) */
                    /* cerr << k << "@@" << endl; */
                /* exit(0); */
            /* } */
        }
        for (int j = 0; j < n; j++) isl[j].clear();
    }
    return ans;
}

void solve() {
    scanf("%s", strbuf);
    s = strbuf;
    scanf("%s", strbuf);
    h = strbuf;

    int n = s.size(), tp = 0;
    for (auto k : h) tp += k - '0';
    if (s == h) return (void)puts("0");
    if (tp == 0) return (void)puts("-1");

    for (int i = 0; i < n; i++) {
        if (h[i] == '1') { tpl[i] = tpr[i] = 0; continue; }
        tpl[i] = tpr[i] = 1;
        while (h[(i - tpl[i] + n) % n] == '0') tpl[i]++;
        while (h[(i + tpr[i]) % n] == '0') tpr[i]++;
    }

    int ans1 = query(n), ans2 = 0;
    reverse(s.begin(), s.end()), reverse(h.begin(), h.end()), swap(tpl, tpr);
    reverse(tpl.begin(), tpl.begin() + n), reverse(tpr.begin(), tpr.begin() + n);
    /* cerr << endl; */
    ans2 = query(n);
    write(min(ans1, ans2)), puts("");
}

bool _edmer;
int main() {
    cerr << (&_stmer - &_edmer) / 1024.0 / 1024.0 << "MB\n";
    int T = read();
    while (T--) solve();
    return 0;
}

相關文章