题解:P10238 [yLCPC2024] F. PANDORA PARADOXXX

离线倒着做,变成连边,以及询问所有连通块直径最大值。

然后是一个经典结论,就是当边权非负时,两个连通块合并,新的直径端点必然是之前两个连通块的四个端点的其中两个,于是你可以使用并查集维护连通块,以及每个连通块的直径端点。合并的时候大力分讨几种即可。

证明:

考虑新的直径,要么在原来的两个连通块内部,要么横跨连通块。在连通块内部的时候,两个端点就是之前两部分的端点。否则,由于树中离每个点最远的点必然在直径端点上,这个就必然成立了。

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

using ll = long long;

const int N = 1e5 + 5, MOD = 1e9 + 7, HSMOD = 1610612741, HSMOD2 = 998244353; // Remember to change

int n, m, q, t, a[N];
mt19937 rnd(chrono::system_clock::now().time_since_epoch().count());

long long qpow(long long a, long long b)
{
	long long res = 1ll, base = a;
	while (b)
	{
		if (b & 1ll) res = res * base % MOD;
		base = base * base % MOD;
		b >>= 1ll;
	}
	return res;
}

bool isprime(int x)
{
	if (x == 1) return 0;
	for (int i = 2; 1ll * i * i <= x; i++) if (x % i == 0) return 0;
	return 1;
}

namespace FastIo
{
	#define QUICKCIN ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
	int read()
	{
		char ch = getchar();
		int x = 0, f = 1;
		while ((ch < '0' || ch > '9') && ch != '-') ch = getchar();
		while (ch == '-')
		{
			f = -f;
			ch = getchar();
		}
		while (ch >= '0' && ch <= '9')
		{
			x = (x << 1) + (x << 3) + (ch ^ 48);
			ch = getchar();
		}
		return x * f;
	}
	template<class T>
	void write(T x)
	{
		if (x < 0)
		{
			putchar('-');
			x = -x;
		}
		if (x > 9) write(x / 10);
		putchar(x % 10 + '0');
	}
	template<class T>
	void writeln(T x)
	{
		write(x);
		putchar('\n');
	}
}

template<typename T>
class Bit
{
public:
	T lowbit(T x)
	{
		return x & -x;
	}
	T tr[N];
	void add(T x, T y)
	{
		while (x < N)
		{
			tr[x] += y;
			x += lowbit(x);
		}
	}
	T query(T x)
	{
		T sum = 0;
		while (x)
		{
			sum += tr[x];
			x -= lowbit(x);
		}
		return sum;
	}
};

int u[N],v[N];
ll w[N];
vector<pair<int,ll>> G[N];
int fa[N][21];
ll dis[N];
int dep[N];

void dfs(int u,int f,int d)
{
	fa[u][0]=f;
	dep[u]=d;
	for(auto&[j,w]:G[u])
	{
		if(j!=f)
		{
			dis[j]=dis[u]+w;
			dfs(j,u,d+1);
		}
	}
}

ll res[N];

inline int LCA(int u,int v)
{
	if(u==v) return u;
	if(dep[u]<dep[v]) swap(u,v);
	int k=dep[u]-dep[v],c=0;
	while(k)
	{
		if(k&1) u=fa[u][c];
		c++;
		k>>=1;
	}
	if(u==v) return u;
	for(int i=20;i>=0;i--) if(fa[u][i]^fa[v][i]) u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}

inline ll getdis(int u,int v)
{
	return dis[u]+dis[v]-2*dis[LCA(u,v)];
}

ll ans=-1ll;

class Union_Find
{
public:
	int fa[N];
	pair<int,int> diamt[N];
	void Init()
	{
		for(int i=1;i<=n;i++) fa[i]=i,diamt[i]=make_pair(i,i);
	}
	int find(int u)
	{
		return (fa[u]==u?u:fa[u]=find(fa[u]));
	}
	void merge(int u,int v)
	{
		if((u=find(u))==(v=find(v))) return;
		vector<int> ver;
		ver.emplace_back(diamt[u].first);
		ver.emplace_back(diamt[u].second);
		ver.emplace_back(diamt[v].first);
		ver.emplace_back(diamt[v].second);
		int ru=0,rv=0;
		ll md=-1ll;
		for(int i=0;i<4;i++)
		{
			for(int j=i+1;j<4;j++)
			{
				ll nd=getdis(ver[i],ver[j]);
				if(nd>md) md=nd,ru=ver[i],rv=ver[j];
			}
		}
		ans=max(ans, md);
		fa[v]=u;
		diamt[u]=make_pair(ru,rv);
	}
}dsu;
bool flag[N];

int main()
{
	ios::sync_with_stdio(0), cin.tie(0);
	cin>>t;
	while(t--)
	{
		cin>>n>>q;
		dsu.Init();
		for(int i=1;i<=n;i++) G[i].clear(),G[i].shrink_to_fit(),flag[i]=0;
		for(int i=1;i<n;i++) 
		{
			cin>>u[i]>>v[i]>>w[i];
			G[u[i]].emplace_back(make_pair(v[i],w[i]));
			G[v[i]].emplace_back(make_pair(u[i],w[i]));
		}
		dfs(1,0,1);
		vector<int> del;
		for(int i=1;i<=q;i++)
		{
			int e;
			cin>>e;
			flag[e]=1;
			del.emplace_back(e);
		}
		ans=0ll;
		reverse(del.begin(),del.end());
		for(int j=1;j<=20;j++) for(int i=1;i<=n;i++) fa[i][j]=fa[fa[i][j-1]][j-1];
		for(int i=1;i<n;i++) if(!flag[i]) dsu.merge(u[i],v[i]);
		int idx=q;
		for(auto&j:del)
		{
			res[idx]=ans;
			idx--;
			dsu.merge(u[j],v[j]);
		}
		for(int i=1;i<=q;i++) cout<<res[i]<<"\n";
	}
	return 0;
}
posted @ 2024-03-14 19:29  HappyBobb  阅读(0)  评论(0编辑  收藏  举报  来源