結果

問題 No.2809 Sort Query
ユーザー keisuke6
提出日時 2024-07-06 14:02:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,228 bytes
コンパイル時間 2,347 ms
コンパイル使用メモリ 216,624 KB
最終ジャッジ日時 2025-02-22 02:25:10
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32 WA * 32 RE * 2 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define int long long
struct SEG{
private:
    int n;
    vector<int> m;
public:
    SEG(){
        n = (1ll<<20);
        m.resize(n);
    }
    void add(int a, int w) {
        for (int x = a; x < n; x |= x + 1) {
            m[x] += w;
        }
    }
    int f(int k){
        int l = 0, now = (1ll<<19)-1, s = 0;
        for(int i=18;i>=0;i--){
            if(s+m[now] < k){
                s += m[now];
                l += (1ll<<(i+1));
                now += (1ll<<(i+1));
            }
            now -= (1ll<<i);
        }
        if(s+m[now] < k) l++;
        return l;
    }
};
signed main(){
    int N,Q;
    cin>>N>>Q;
    vector<int> A(N);
    for(int i=0;i<N;i++) cin>>A[i];
    set<int> s;
    for(int x:A) s.insert(x);
    vector<int> X(Q),Y(Q),Z(Q);
    for(int i=0;i<Q;i++){
        cin>>X[i];
        if(X[i] == 1){
            cin>>Y[i]>>Z[i];
            s.insert(Z[i]);
        } 
        if(X[i] == 3) cin>>Y[i];
    }
    unordered_map<int,int> mp,mmp;
    int inda = 0;
    for(int x:s){
        mp[inda] = x;
        mmp[x] = inda;
        inda++;
    }
    for(int i=0;i<N;i++) A[i] = mmp[A[i]];
    for(int i=0;i<Q;i++)if(X[i] == 1) Z[i] = mmp[Z[i]];
    reverse(X.begin(),X.end());
    reverse(Y.begin(),Y.end());
    reverse(Z.begin(),Z.end());
    while(Q--){
        int t = X[Q];
        if(t == 1){
            int a = Y[Q], b = Z[Q];
            A[a-1] = b;
        }
        else if(t == 2){
            sort(A.begin(),A.end());
            break;
        }
        else{
            int a = Y[Q];
            cout<<mp[A[a-1]]<<endl;
        }
    }
    SEG seg;
    unordered_map<int,int> m;
    for(int i=0;i<N;i++) seg.add(A[i],1);
    while(Q--){
        int t = X[Q];
        if(t == 1){
            int a = Y[Q], b = Z[Q];
            m[a] = b;
        }
        else if(t == 2){
            unordered_map<int,int> mm;
            for(auto [a,b]:m){
                seg.add(seg.f(a),-1);
                seg.add(b,1);
                A[a] = b;
            }
            m = mm;
        }
        else{
            int a = Y[Q];
            if(m.count(a)) cout<<mp[m[a]]<<endl;
            else cout<<mp[seg.f(a)]<<endl;
        }
    }
}
0