結果

問題 No.2809 Sort Query
ユーザー highlighterhighlighter
提出日時 2024-06-12 20:22:14
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,366 bytes
コンパイル時間 3,427 ms
コンパイル使用メモリ 266,212 KB
実行使用メモリ 41,148 KB
最終ジャッジ日時 2024-07-12 21:05:33
合計ジャッジ時間 31,811 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
testcase_51 WA -
testcase_52 WA -
testcase_53 WA -
testcase_54 WA -
testcase_55 WA -
testcase_56 WA -
testcase_57 WA -
testcase_58 WA -
testcase_59 WA -
testcase_60 WA -
testcase_61 WA -
testcase_62 WA -
testcase_63 WA -
testcase_64 WA -
testcase_65 WA -
testcase_66 WA -
testcase_67 WA -
testcase_68 WA -
testcase_69 WA -
testcase_70 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

struct Fenwick_tree{
	private:
	unsigned int siz=1;
	vector<int> dat;
	public:
	Fenwick_tree(int N) : dat(N,0){
		siz=N;
	}
	void add(int k,int x){
		for(int i=k;i<siz;i+=((i+1)&(-i-1))){
			dat[i]+=x;
		}
	}
	int max_right(int x){
		int ans=0;
		int res=0;
		int h=bit_floor(siz-1);
		while(h>0){
			if(ans+h<=siz){
				if(res+dat[ans+h-1]<=x){
					res+=dat[ans+h-1];
					ans+=h;
				}
			}
			h/=2;
		}
		return ans;
	}
};

signed main(){
	int N,Q;
	scanf("%d%d",&N,&Q);
	vector<long long> A(N);
	for(int i=0;i<N;i++){
		scanf("%lld",&A[i]);
	}
	vector<long long> data=A;
	vector<vector<long long>> query(Q);
	for(int i=0;i<Q;i++){
		int c;
		scanf("%d",&c);
		if(c==1){
			int k;
			long long x;
			scanf("%d%lld",&k,&x);
			data.emplace_back(x);
			query[i]={c,k,x};
			continue;
		}
		if(c==2){
			query[i]={c};
			continue;
		}
		int k;
		scanf("%d",&k);
		query[i]={c,k};
	}
	vector<long long> data2=data;
	sort(data2.begin(),data2.end());
	data2.erase(unique(data2.begin(),data2.end()),data2.end());
	for(long long &i : data){
		i=lower_bound(data2.begin(),data2.end(),i)-data2.begin();
	}
	vector<long long> relation(*max_element(data.begin(),data.end())+1);
	for(int i=0;i<N;i++){
		relation[data[i]]=A[i];
		A[i]=data[i];
	}
	int mem=N;
	for(int i=0;i<Q;i++){
		if(query[i][0]==1){
			relation[data[mem]]=query[i][2];
			query[i][2]=data[mem];
			mem++;
		}
	}
	queue<long long> change;
	vector<bool> change2(N,false);
	Fenwick_tree tree(relation.size());
	for(int i=0;i<N;i++){
		tree.add(A[i],1);
		change.emplace(i);
	}
	for(int i=0;i<Q;i++){
		if(query[i][0]==1){
			int k=query[i][1]-1;
			int x=query[i][2];
			A[k]=x;
			if(!change2[k]){
				change.emplace(k);
				change2[k]=true;
			}
		}
		if(query[i][0]==2){
			int t=change.size();
			for(;t--;){
				long long j=change.front();
				change2[j]=false;
				change.pop();
				long long t=tree.max_right(j);
				change.emplace((long long)(N)*t+j);
			}
			while(!change.empty()){
				int t=change.front()/N;
				int j=change.front()%N;
				if(A[j]==-1){
					change.pop();
					continue;
				}
				tree.add(t,-1);
				tree.add(A[j],1);
				A[j]=-1;
				change.pop();
			}
		}
		if(query[i][0]==3){
			int k=query[i][1]-1;
			if(A[k]!=-1){
				printf("%d\n",relation[A[k]]);
			}
			else{
				printf("%d\n",relation[tree.max_right(k)]);
			}
		}
	}
}
0