結果

問題 No.1526 Sum of Mex 2
ユーザー t33ft33f
提出日時 2021-06-02 00:00:13
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,123 bytes
コンパイル時間 1,031 ms
コンパイル使用メモリ 93,024 KB
実行使用メモリ 11,472 KB
最終ジャッジ日時 2024-04-26 09:56:39
合計ジャッジ時間 3,306 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 7 ms
5,376 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 37 ms
8,064 KB
testcase_17 AC 27 ms
6,656 KB
testcase_18 AC 5 ms
5,376 KB
testcase_19 WA -
testcase_20 AC 17 ms
5,632 KB
testcase_21 WA -
testcase_22 AC 23 ms
6,272 KB
testcase_23 AC 38 ms
7,808 KB
testcase_24 AC 39 ms
8,064 KB
testcase_25 AC 36 ms
7,808 KB
testcase_26 AC 37 ms
8,064 KB
testcase_27 AC 39 ms
8,064 KB
testcase_28 AC 41 ms
8,192 KB
testcase_29 AC 37 ms
8,320 KB
testcase_30 AC 38 ms
8,192 KB
testcase_31 AC 40 ms
8,320 KB
testcase_32 AC 37 ms
7,936 KB
testcase_33 WA -
evil_largest WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <tuple>
#include <algorithm>
#include <set>
#include <vector>
#include <iostream>
using namespace std;
int main() {
    int n; cin >> n;
    int a[n]; for (int i = 0; i < n; i++) cin >> a[i];
    vector<int> b[n+1];
    for (int i = 0; i < n; i++) b[a[i]].push_back(i);
    long long ans = (long long)n * (n + 1) / 2;
    if (b[1].empty()) {
        cout << ans << endl;
        return 0;
    }
    set<tuple<int, int, long long> > v;
    long long s = 0;
    for (int i : b[1]) {
        const int l = v.empty() ? -1 : get<0>(*v.rbegin());
        const long long c = (long long)(i - l) * (n - i);
        v.insert({i, i, c});
        s += c;
    }
    ans += s;
    for (int x = 2; x <= n; x++) {
        // cerr << "x = " << x << endl;
        // for (auto [i, j, k] : v) cerr << i << ' ' << j << "; "; cerr << endl;
        // cerr << ans << endl;
        if (b[x].empty()) break;
        auto left = v.begin();
        for (int k : b[x]) {
            // cerr << "k = " << k << endl;
            // for (auto [i, j, k] : v) cerr << i << ' ' << j << "; "; cerr << endl;
            auto it = v.lower_bound({k, k, 0});
            // if (it == v.end()) cerr << "end\n";
            if (it != left) {
                while (next(left) != v.end() && get<1>(*next(left)) < k) {
                    s -= get<2>(*left);
                    left = v.erase(left);
                }
                const int l1 = left == v.begin() ? -1 : get<0>(*prev(left));
                const long long c = (long long)(get<0>(*left) - l1) * (n - k);
                s += c;
                v.insert({get<0>(*left), k, c});
                s -= get<2>(*left);
                v.erase(left);
            }
            if (it != v.end()) {
                const int r = get<1>(*it);
                const int l1 = it == v.begin() ? -1 : get<0>(*prev(it));
                const long long c = (long long)(k - l1) * (n - r);
                s += c;
                v.insert({k, r, c});
                // cerr << k << ' ' << r << endl;
            }
            left = it;
        }
        ans += s;
    }

    cout << ans << endl;
}
0