結果

問題 No.2311 [Cherry 5th Tune] Cherry Month
ユーザー kotatsugame
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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";
}
0