結果
問題 | No.2311 [Cherry 5th Tune] Cherry Month |
ユーザー |
|
提出日時 | 2023-05-19 23:00:46 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 537 ms / 4,600 ms |
コード長 | 2,072 bytes |
コンパイル時間 | 1,120 ms |
コンパイル使用メモリ | 85,416 KB |
実行使用メモリ | 66,556 KB |
最終ジャッジ日時 | 2024-06-12 05:55:02 |
合計ジャッジ時間 | 29,163 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 51 |
ソースコード
#include<iostream>#include<vector>#include<algorithm>#include<atcoder/fenwicktree>using namespace std;int N,T,Q;long A[150000];vector<pair<int,int> >G[150000];vector<pair<int,int> >H[150000];vector<int>Tm[150000];atcoder::fenwick_tree<long>Bit[150000];long ans[4<<17];int op[4<<17],X[4<<17],Y[4<<17];vector<int>qids[4<<17];vector<int>Vs[150000];int pr[150000];long del[150000];int qU[4<<17];int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin>>N;for(int i=0;i<N;i++)cin>>A[i];cin>>T;for(int i=1;i<=T;i++){cin>>op[i]>>X[i]>>Y[i];X[i]--;if(op[i]==1){Y[i]--;G[X[i]].push_back(make_pair(Y[i],i));G[Y[i]].push_back(make_pair(X[i],i));}else if(op[i]==3){Tm[X[i]].push_back(i);}}for(int i=0;i<N;i++){sort(Tm[i].begin(),Tm[i].end());Bit[i]=atcoder::fenwick_tree<long>(Tm[i].size());for(pair<int,int>e:G[i])if(make_pair(G[i].size(),i)<make_pair(G[e.first].size(),e.first)){H[i].push_back(e);}Vs[i].push_back(i);pr[i]=i;del[i]=0;}cin>>Q;for(int i=0;i<Q;i++){int s;cin>>s>>qU[i];qU[i]--;qids[s].push_back(i);}for(int i=0;i<=T;i++){if(i>0){if(op[i]==1){int u=pr[X[i]],v=pr[Y[i]];if(u!=v){if(Vs[u].size()<Vs[v].size())swap(u,v);for(int w:Vs[v]){Vs[u].push_back(w);pr[w]=u;A[w]-=del[v];A[w]+=del[u];}}}else if(op[i]==2)A[X[i]]-=Y[i];else if(op[i]==3){int u=X[i];Bit[u].add(lower_bound(Tm[u].begin(),Tm[u].end(),i)-Tm[u].begin(),Y[i]);A[u]-=Y[i];for(pair<int,int>e:H[u])if(e.second<=i){A[e.first]-=Y[i];}}else{del[pr[X[i]]]+=Y[i];}}for(int qid:qids[i]){int u=qU[qid];long nowans=A[u]-del[pr[u]];for(pair<int,int>e:H[u])if(e.second<=i){int v=e.first;int id=lower_bound(Tm[v].begin(),Tm[v].end(),e.second)-Tm[v].begin();nowans-=Bit[e.first].sum(id,Tm[e.first].size());}ans[qid]=max(nowans,0L);}}for(int i=0;i<Q;i++)cout<<ans[i]<<"\n";}