結果

問題 No.1437 01 Sort
コンテスト
ユーザー novaandcabral
提出日時 2024-04-22 14:39:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,229 bytes
コンパイル時間 1,775 ms
コンパイル使用メモリ 178,592 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-14 13:45:36
合計ジャッジ時間 2,731 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 4 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

unsigned long long solve(int N, string S) {
    if (vector<char>(S.begin(), S.end()) == vector<char>(S.begin(), S.end())) {
        return 0;
    }

    string S2 = S + S;

    // 计算 S[l:r] 中 '0' 的数量，zero_cnt[r] - zero_cnt[l]
    vector<int> zero_cnt(S2.length() + 1, 0);
    vector<int> one_cnt(S2.length() + 1, 0);
    for (char c : S2) {
        zero_cnt.push_back(zero_cnt.back() + (c == '0'));
        one_cnt.push_back(one_cnt.back() + (c == '1'));
    }

    unordered_map<string, int> cache;

    function<int(int, int, int)> calc = [&](int l, int t, int r) -> int {
        int left0 = zero_cnt[t + 1] - zero_cnt[l];
        int right1 = one_cnt[r + 1] - one_cnt[t + 1];
        int end = t + right1;
        int rem = (N - 1 - end) % N;
        assert(l <= t && t <= end && end <= r);
        if (end < N) {
            if (right1) {
                int rskip = (one_cnt[N + 1] - one_cnt[t + 1]) ? 1 : 0;
                right1 = right1 + 1 - rskip;
            }
            return max(left0, right1) * N + rem;
        } else if (t < N) {
            int rskip = (one_cnt[N + 1] - one_cnt[t + 1]) ? 1 : 0;
            return max(left0 - 1, right1 - rskip) * N + rem;
        } else {
            int lskip = (zero_cnt[N + 1] - zero_cnt[l + 1]) ? 1 : 0;
            return max(left0 - lskip, right1) * N + rem;
        }
    };

    vector<int> ones;
    for (int i = 0; i < S2.length(); i++) {
        if (S2[i] == '1') {
            ones.push_back(i);
        }
    }

    unsigned long long answer = 1e18;
    int t = -1;
    for (int i = 0; i < one_cnt[N]; i++) {
        int l = ones[i];
        int r = ones[i + one_cnt[N] - 1];
        t = max(l, t);
        
            int c = calc(l, t, r);
        while (true) {
            int nt = t + 1;
            while (nt <= r && c == calc(l, nt, r)) {
                nt++;
            }
            if (nt == r + 1 || c < calc(l, nt, r)) {
                break;
            }
            t = nt;
        }
        answer = min(answer, (unsigned long long)c);
    }
    return answer;
}

int main() {
    int N;
    cin >> N;
    string S;
    cin >> S;
    cout << solve(N, S) << endl;
    return 0;
}
0