题目链接
小号终于上蓝了,嘿嘿。这个题比赛的时候一直在想假算法换根dp,赛后看题解没想到求联通块就完事了qwq。
题意:给你一颗树,每条边都有一个权值。q次询问,每次询问一个qi,问这颗树有多少条路径,路径上边权最大值小于qi。
题解:因为是离线查询,所以我们可以考虑离线做法。先将qi从小到大排序,然后依次更新答案。每次只将边权小于qi的边添加进图里,为啥要这样做呢,这样就能保证图里任意路径的边权最大值都小于qi。于是我们会得到一个个联通块,对于一个联通块能贡献的答案是 sz*(sz-1)/2;联通块的更新就可以用到并查集,再用sz保存每个联通块点的个数。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5+100;
int n,m;
struct node{
int u,v,w;
}e[N];
struct Q{
int val,id;
}q[N];
bool cmp1(node a,node b){
return a.w<b.w;
}
bool cmp2(Q a,Q b){
return a.val<b.val;
}
int pre[N],sz[N];
ll ans[N];
void find(int& x){
while(x!=pre[x])
x=pre[x]=pre[pre[x]];
}
void gkd(){
for(int i=1;i<=n;i++) pre[i]=i,sz[i]=1;
ll tmp=0;
int id=1;
for(int i=1;i<=m;i++){
while(e[id].w<=q[i].val&&id<n){
int x = e[id].u;
int y = e[id].v;
find(x);find(y);
if(x!=y){
tmp-=1ll*(sz[x]-1)*sz[x]/2;
tmp-=1ll*(sz[y]-1)*sz[y]/2;
pre[y]=x;sz[x]+=sz[y];
tmp+=1ll*(sz[x]-1)*sz[x]/2;
}
id++;
}
ans[q[i].id]=tmp;
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<n;++i){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
e[i]=node{u,v,w};
}
for(int i=1;i<=m;i++){
scanf("%d",&q[i].val);
q[i].id=i;
}
sort(e+1,e+n,cmp1);
sort(q+1,q+1+m,cmp2);
gkd();
for(int i=1;i<=m;++i){
if(i!=m) printf("%lld ",ans[i]);
else printf("%lld\n",ans[i]);
}
}