結果

問題 No.1437 01 Sort
ユーザー novaandcabralnovaandcabral
提出日時 2024-04-22 14:33:53
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,724 bytes
コンパイル時間 2,415 ms
コンパイル使用メモリ 172,552 KB
実行使用メモリ 11,764 KB
最終ジャッジ日時 2024-04-22 14:33:59
合計ジャッジ時間 5,849 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 AC 2 ms
5,376 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 AC 2 ms
5,376 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 TLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int f(int l, int ll, int n, string s, vector<int>& bit, vector<int>& out) {
    int one = s.size() - count(s.begin(), s.end(), '0');
    int ans = 0;

    int sp_l = (l + one) % n;
    int sp_r = n;
    if (sp_l == 0) sp_l = n;

    int ansl = 0;
    if (bit[ll] <= l + n) {
        if (sp_l <= ((bit[ll] - 1) % n + 1) && ((bit[ll] - 1) % n + 1) <= sp_r && out[bit[ll] % n] == 0)
            ansl = l + n - bit[ll] - 1;
        else
            ansl = l + n - bit[ll];
    }

    int ansr = 0;
    if (l + n + one - 1 < bit[ll + one - 1]) {
        int ng = -1;
        int ok = one;
        while (ng + 1 != ok) {
            int mid = (ng + ok) / 2;
            if (l + n + mid < bit[ll + mid])
                ok = mid;
            else
                ng = mid;
        }

        bool flag1 = true;
        bool flag2 = false;
        int x = l + n + ok;
        if (x % n != 0) x += n - (x % n);
        if (x <= l + n + one - 1) {
            if (l + n + ok <= bit[ll + ok] && bit[ll + ok] <= x)
                flag2 = true;
            x += n;
        }
        if (l + n + ok <= bit[ll + ok] && bit[ll + ok] <= x)
            flag1 = false;
        ansr = one - ok + flag1 - flag2;
    }
    return max(ansl, ansr);
}

signed main() {
    int n;
    cin >> n;
    string s;
    cin >> s;
    int ans = 1e10;

    int zero = count(s.begin(), s.end(), '0');
    int one = n - zero;
    if (zero == 0 || zero == n) {
        cout << 0 << endl;
        return 0;
    }

    vector<int> out(n, 0);
    if (s[0] == '1') {
        out[0] = 1;
        for (int i = n - 1; i >= 0; --i) {
            if (s[i] == '1')
                out[i] = 1;
            else
                break;
        }
    }

    vector<int> bit;
    for (int i = 0; i < n; ++i) {
        if (s[i] == '1') {
            bit.push_back(i);
            bit.push_back(i + n);
            bit.push_back(i + 2 * n);
            bit.push_back(i + 3 * n);
        }
    }
    sort(bit.begin(), bit.end());

    for (int l = 0; l < n; ++l) {
        int anss = (-(l + one)) % n;
        int sp_l = (l + one) % n;
        int sp_r = n;
        if (sp_l == 0) sp_l = n;
        
        int ng = -1;
        int ok = n;
        while (ng + 1 != ok) {
            int mid = (ng + ok) / 2;
            int x = l + n - mid - 1;
            if (!(sp_l <= ((x - 1) % n + 1) && out[x % n] == 0)) x += 1;
            int ll = lower_bound(bit.begin(), bit.end(), x) - bit.begin();

            if (f(l, ll, n, s, bit, out) <= mid)
                ok = mid;
            else
                ng = mid;
        }
        ans = min(ans, anss + ok * n);
    }

    cout << ans << endl;
    return 0;
}
0