题目链接:P4185 [USACO18JAN]MooTube G - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
在业余时间,Farmer John 创建了一个新的视频共享服务,他将其命名为 MooTube。在 MooTube 上,Farmer John 的奶牛可以录制,分享和发现许多有趣的视频。他的奶牛已经发布了 N 个视频(1≤N≤105),为了方便将其编号为 1…N 。然而,FJ 无法弄清楚如何帮助他的奶牛找到他们可能喜欢的新视频。
FJ 希望为每个 MooTube 视频创建一个“推荐视频”列表。这样,奶牛将被推荐与他们已经观看过的视频最相关的视频。
FJ 设计了一个“相关性”度量标准,顾名思义,它确定了两个视频相互之间的相关性。他选择N−1 对视频并手动计算其之间的相关性。然后,FJ 将他的视频建成一棵树,其中每个视频是节点,并且他手动将 N−1 对视频连接。为了方便,FJ 选择了N−1 对,这样任意视频都可以通过一条连通路径到达任意其他视频。 FJ 决定将任意一对视频的相关性定义为沿此路径的任何连接的最小相关性。
Farmer John 想要选择一个 K 值,以便在任何给定的 MooTube 视频旁边,推荐所有其他与该视频至少有 K 相关的视频。然而,FJ 担心会向他的奶牛推荐太多的视频,这可能会分散他们对产奶的注意力!因此,他想设定适当的 K 值。 Farmer John希望得到您的帮助,回答有关 K 值的推荐视频的一些问题。
![](https://i-blog.csdnimg.cn/blog_migrate/ef38287b8d2313fe45371bd18f179cf7.png)
将关系按照价值从大到小排序,如果当前某个值对于k满足的话,那么在他之前的关系都是大于等于k的;同时对查询也进行排序,从大到小排序,如果大的k能够符合的话,那么也可以符合小的k
//并查集
int find(int x){
if(f[x]==x) return x;
else return f[x]=find(f[x]);
}
for(int i=1;i<=Q;i++){
while(s[cnt].w>=q[i].k&&cnt<=n){
int x=find(s[cnt].u);
int y=find(s[cnt].v);
cnt++;
if(x==y) continue;
f[x]=y;
Size[y]+=Size[x];//代表当前点的连通块中的成员数量
}
ans[q[i].id]=Size[find(q[i].v)]-1;//id标记了答案的顺序
}
对于某一个答案,从cnt开始遍历,如果满足当前的w是大于题目所需要的k值,说明当前这条关系的可以用的,将二者联系起来,同时在建立父子关系的时候将size传给父亲。该次查询的答案就是该点(或者其父亲)的size大小,-1是去掉本身
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
#define ll long long
#define D double
#define maxn 100005
using namespace std;
const int N=100005;
struct node{
int u,v,w;
}s[N];
struct query{
int k,v,id;
}q[N];
int n,Q;
int f[N],Size[N],ans[N];
bool cmp1(node a,node b){
return a.w>b.w;
}
bool cmp2(query a,query b){
return a.k>b.k;
}
int find(int x){
if(f[x]==x) return x;
else return f[x]=find(f[x]);
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
cin>>n>>Q;
for(int i=1;i<=n;i++){
f[i]=i;
Size[i]=1;
}
for(int i=1;i<n;i++){
cin>>s[i].u>>s[i].v>>s[i].w;
}
sort(s+1,s+n,cmp1);
for(int i=1;i<=Q;i++){
cin>>q[i].k>>q[i].v;
q[i].id=i;
}
sort(q+1,q+Q+1,cmp2);
int cnt=1;
for(int i=1;i<=Q;i++){
while(s[cnt].w>=q[i].k&&cnt<=n){
int x=find(s[cnt].u);
int y=find(s[cnt].v);
cnt++;
if(x==y) continue;
f[x]=y;
Size[y]+=Size[x];
}
ans[q[i].id]=Size[find(q[i].v)]-1;
}
for(int i=1;i<=Q;i++){
cout<<ans[i]<<endl;
}
return 0;
}