結果

問題 No.1788 Same Set
ユーザー hitonanodehitonanode
提出日時 2021-12-17 01:15:31
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 908 ms / 4,000 ms
コード長 2,298 bytes
コンパイル時間 1,723 ms
コンパイル使用メモリ 138,720 KB
実行使用メモリ 75,404 KB
最終ジャッジ日時 2024-09-14 01:23:26
合計ジャッジ時間 21,348 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 3 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 3 ms
6,940 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
6,944 KB
testcase_11 AC 115 ms
16,284 KB
testcase_12 AC 268 ms
24,436 KB
testcase_13 AC 325 ms
26,852 KB
testcase_14 AC 450 ms
28,960 KB
testcase_15 AC 590 ms
29,408 KB
testcase_16 AC 624 ms
29,836 KB
testcase_17 AC 842 ms
60,244 KB
testcase_18 AC 800 ms
47,528 KB
testcase_19 AC 700 ms
75,404 KB
testcase_20 AC 185 ms
16,500 KB
testcase_21 AC 624 ms
29,328 KB
testcase_22 AC 663 ms
29,748 KB
testcase_23 AC 502 ms
26,628 KB
testcase_24 AC 591 ms
29,008 KB
testcase_25 AC 763 ms
32,860 KB
testcase_26 AC 760 ms
32,608 KB
testcase_27 AC 283 ms
24,052 KB
testcase_28 AC 821 ms
37,948 KB
testcase_29 AC 825 ms
37,352 KB
testcase_30 AC 295 ms
24,072 KB
testcase_31 AC 838 ms
37,596 KB
testcase_32 AC 860 ms
46,276 KB
testcase_33 AC 876 ms
46,476 KB
testcase_34 AC 908 ms
46,488 KB
testcase_35 AC 896 ms
47,556 KB
testcase_36 AC 888 ms
47,584 KB
testcase_37 AC 866 ms
47,488 KB
testcase_38 AC 875 ms
47,436 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <unordered_map>
#include <utility>
#include <vector>
using namespace std;

#include <atcoder/lazysegtree>
using S = pair<int, int>;  // min, num of min
S op(S l, S r) {
    if (l.first < r.first) return l;
    if (l.first > r.first) return r;
    return {l.first, l.second + r.second};
}
S e() { return {1 << 20, 0}; }
using F = int;
S mapping(F f, S x) { return {x.first + f, x.second}; }
F composition(F f, F g) { return f + g; }
F id() { return 0; }

int main() {
    int N;
    cin >> N;

    vector<vector<tuple<int, int, int>>> updates(N + 1);

    unordered_map<int, pair<vector<int>, vector<int>>> mp;
    for (int i = 0; i < N; ++i) {
        int a;
        cin >> a;
        mp[a].first.push_back(i);
    }
    for (int i = 0; i < N; ++i) {
        int b;
        cin >> b;
        mp[b].second.push_back(i);
    }

    for (const auto &[key, xaxb] : mp) {
        const auto &[xa, xb] = xaxb;
        int ia = 0, ib = 0, l = 0;
        while (ia < int(xa.size()) or ib < int(xb.size())) {
            if (ia == xa.size()) {
                updates[l].emplace_back(xb[ib], N, 1);
                updates[xb[ib] + 1].emplace_back(xb[ib], N, -1);
                ib++;
            } else if (ib == xb.size()) {
                updates[l].emplace_back(xa[ia], N, 1);
                updates[xa[ia] + 1].emplace_back(xa[ia], N, -1);
                ia++;
            } else if (xa[ia] == xb[ib]) {
                l = xa[ia] + 1;
                ia++;
                ib++;
            } else if (xa[ia] < xb[ib]) {
                updates[l].emplace_back(xa[ia], xb[ib], 1);
                updates[xa[ia] + 1].emplace_back(xa[ia], xb[ib], -1);
                l = xa[ia] + 1;
                ia++;
            } else {
                updates[l].emplace_back(xb[ib], xa[ia], 1);
                updates[xb[ib] + 1].emplace_back(xb[ib], xa[ia], -1);
                l = xb[ib] + 1;
                ib++;
            }
        }
    }

    long long ret = 0;
    atcoder::lazy_segtree<S, op, e, F, mapping, composition, id> tree(vector<S>(N, S(0, 1)));
    for (int i = 0; i < N; ++i) {
        for (auto [l, r, v] : updates[i]) tree.apply(l, r, v);
        auto pr = tree.prod(i, N);
        if (pr.first == 0) ret += pr.second;
    }
    cout << ret << '\n';
}
0