题解 洛谷 P4185(P6111) 【[USACO18JAN]MooTube G(S)】

这题的加强版的我们 PJ 模拟赛的 T4，个人感觉还有点意思的。

先把题目转化一下，就是每个询问给你一个 \(K\) 值和 \(v\) 值，把所有小于 \(K\) 的边都删掉，然后从 \(v\) 能走到几个节点。

可以离线，所以选择离线。把问题按 \(K\) 从大到小排序，把边按权值从大到小排序。然后依次考虑每个问题，进行加边操作。这里加边的时候要维护每个点所在的连通块的大小，所以可以用权值并查集维护。

为什么不从小到大排序然后维护删边？因为删边比加边难维护多了。

#include<bits/stdc++.h>
using namespace std;
int n,m;
struct edge//每条边
{
	int x,y,z;
}t[100005];
struct ques//每个问题
{
	int k,x,id;
}q[100005];
bool cmp1(edge x,edge y)
{
	return x.z>y.z;
}
bool cmp2(ques x,ques y)
{
	return x.k>y.k;
}
int fa[100005],res[100005],ans[100005];//res表示i所在的连通块的大小
int find(int x)
{
	if (x==fa[x]) return x;
	int u=find(fa[x]);
	res[x]=res[u];
	fa[x]=u;
	return fa[x];
}
int main()
{
    cin >> n >> m;
    for (int i=1;i<n;i++)
    {
    	cin >> t[i].x >> t[i].y >> t[i].z;
	}
	for (int i=1;i<=n;i++) fa[i]=i,res[i]=1;
	sort(t+1,t+n,cmp1);
	for (int i=1;i<=m;i++)
	{
		cin >> q[i].k >> q[i].x;
		q[i].id=i;
	}
	sort(q+1,q+m+1,cmp2);
	int num=0;
	for (int i=1;i<=m;i++)
	{
		for (;num<n;num++)
		{
			if (t[num+1].z>=q[i].k) 
			{
				int x1=find(t[num+1].x),y1=find(t[num+1].y);
				fa[x1]=y1;
				res[x1]+=res[y1];
				res[y1]=res[x1];
			}
			else break;
		}
		int p=find(q[i].x);
		ans[q[i].id]=res[q[i].x]-1;
	}
	for (int i=1;i<=m;i++) cout << ans[i] << endl;
    return 0;
}