結果

問題 No.3269 Leq-K Partition
ユーザー rabimea
提出日時 2025-09-23 19:54:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 530 ms / 6,000 ms
コード長 2,021 bytes
コンパイル時間 4,166 ms
コンパイル使用メモリ 287,080 KB
実行使用メモリ 82,720 KB
最終ジャッジ日時 2025-09-23 19:54:59
合計ジャッジ時間 11,719 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

// #pragma GCC optimize ("Ofast")
// #pragma GCC optimize ("unroll-loops")
// #pragma GCC target ("avx,avx2,fma")
#include <bits/stdc++.h>

using std::cin, std::cout, std::cerr;
using ll = long long;

int LowBit(int x) { return x & (-x); }
struct BIT {
    int n;
    std::vector<int> a;
    BIT(int n) { Init(n); }
    void Init(int n) { this->n = n; a.assign(n + 1, 0); }

    void Update(int p, int x) {
        for(int i = p + 1; i <= n + 1; i += LowBit(i))
            a[i - 1] += x;
    }
    int Query(int p) {
        int r = 0;
        for(int i = p + 1; i; i -= LowBit(i))
            r += a[i - 1];
        return r;
    }
};

int main() {
    std::ios::sync_with_stdio(false);
    int n; cin >> n;
    std::vector<int> a(n + 1);
    std::vector<std::deque<int>> pos(n + 1);
    for(int i = 1; i <= n; i ++) {
        cin >> a[i];
        pos[a[i]].push_back(i);
    }
    BIT bit(n);
    for(int i = 1; i <= n; i ++) {
        if(!pos[i].empty())
            bit.Update(pos[i][0], 1);
    }
    std::vector<int> ans(n + 1);
    std::vector<std::vector<int>> query(n + 1);
    for(int i = 1; i <= n; i ++)
        query[0].push_back(i);
    
    auto check = [&](int p, int k) {
        return bit.Query(p) <= k;
    };

    for(int i = 0; i < n; i ++) {
        // cout << "--------\n";
        // for(int i = 1; i <= n; i ++)
        //     cout << bit.Query(i) << ' ';
        // cout << '\n';
        for(int q : query[i]) {
            int l = i + 1, r = n + 1;
            while(l + 1 < r) {
                int m = (l + r) / 2;
                if(check(m, q))
                    l = m;
                else
                    r = m;
            }
            // cout << i << ' ' << q << ' ' << l << '\n';
            ans[q] ++;
            query[l].push_back(q);
        }
        int x = a[i + 1];
        bit.Update(pos[x][0], -1);
        pos[x].pop_front();
        if(!pos[x].empty())
            bit.Update(pos[x][0], 1);
    }

    for(int i = 1; i <= n; i ++)
        cout << ans[i] << '\n';
}
0