結果

問題 No.2296 Union Path Query (Hard)
ユーザー kotatsugamekotatsugame
提出日時 2023-05-05 23:20:02
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,628 bytes
コンパイル時間 1,063 ms
コンパイル使用メモリ 84,228 KB
実行使用メモリ 50,048 KB
最終ジャッジ日時 2024-05-02 18:33:14
合計ジャッジ時間 12,555 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 17 ms
28,160 KB
testcase_01 AC 52 ms
50,048 KB
testcase_02 AC 52 ms
49,920 KB
testcase_03 AC 51 ms
50,048 KB
testcase_04 TLE -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
testcase_47 -- -
testcase_48 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<set>
#include<cassert>
using namespace std;
int N,X,Q;
vector<pair<int,int> >G[2<<17];
int par[2<<17];
int prD[2<<17];
int pr[18][2<<17],depth[2<<17];
int myrank[2<<17];
long long D[2<<17];
set<pair<long long,int> >mD[2<<17];
vector<int>V[2<<17];
long long ans[2<<17];
long long dfs(int u,int p,int dep,long long dist,int prd,int AP)
{
	par[u]=AP;
	pr[0][u]=p;
	for(int k=1;k<18;k++)
	{
		if(pr[k-1][u]==-1)pr[k][u]=-1;
		else pr[k][u]=pr[k-1][pr[k-1][u]];
	}
	depth[u]=dep;
	D[u]=dist;
	prD[u]=prd;
	V[AP].push_back(u);
	mD[u].clear();
	long long md=0;
	for(pair<int,int>e:G[u])if(e.first!=p)
	{
		long long d=dfs(e.first,u,dep+1,dist+e.second,e.second,AP)+e.second;
		mD[u].insert(make_pair(d,e.first));
		md=max(md,d);
	}
	return md;
}
int lca(int u,int v)
{
	if(depth[u]>depth[v])swap(u,v);
	for(int k=0;k<18;k++)if(depth[v]-depth[u]>>k&1)v=pr[k][v];
	if(u==v)return u;
	for(int k=18;k--;)if(pr[k][u]!=pr[k][v])
	{
		u=pr[k][u];
		v=pr[k][v];
	}
	return pr[0][u];
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N>>X>>Q;
	for(int i=0;i<N;i++)
	{
		V[i].push_back(i);
		for(int k=0;k<18;k++)pr[k][i]=-1;
		myrank[i]=1;
		par[i]=i;
	}
	for(;Q--;)
	{
		int op;cin>>op;
		if(op==1)
		{
			int v,w;cin>>v>>w;
			int u=X;
			if(myrank[par[u]]>myrank[par[v]])swap(u,v);
			myrank[par[u]]=max(myrank[par[u]],myrank[par[v]]+depth[u]+1);
			ans[par[u]]=max(ans[par[u]],ans[par[v]]);
			long long nd=dfs(v,u,depth[u]+1,D[u]+w,w,par[u])+w;
			if(mD[u].empty())
			{
				ans[par[u]]=max(ans[par[u]],nd);
			}
			else
			{
				auto it=mD[u].end();it--;
				ans[par[u]]=max(ans[par[u]],it->first+nd);
			}
			mD[u].insert(make_pair(nd,v));
			{
				int x=u;
				while(pr[0][x]!=-1)
				{
					int y=pr[0][x];
					auto it=mD[x].end();it--;
					while(!mD[y].empty()&&mD[y].rbegin()->second==x)
					{
						auto jt=mD[y].end();jt--;
						mD[y].erase(jt);
					}
					if(mD[y].empty())
					{
						ans[par[u]]=max(ans[par[u]],it->first+prD[x]);
					}
					else
					{
						auto jt=mD[y].end();
						jt--;
						assert(jt->second!=x);
						ans[par[u]]=max(ans[par[u]],it->first+prD[x]+jt->first);
					}
					mD[y].insert(make_pair(it->first+prD[x],x));
					x=y;
				}
			}
			G[u].push_back(make_pair(v,w));
			G[v].push_back(make_pair(u,w));
		}
		else if(op==2)
		{
			int u,v;cin>>u>>v;
			if(par[u]!=par[v])cout<<"-1\n";
			else
			{
				int w=lca(u,v);
				long long d=D[u]+D[v]-2*D[w];
				cout<<d<<"\n";
				X=(X+d%N)%N;
			}
		}
		else if(op==3)
		{
			int v;cin>>v;
			v=par[v];
			cout<<ans[v]<<"\n";
		}
		else
		{
			int v;cin>>v;
			X=(X+v)%N;
		}
	}
}
0