题解CF444E DZY Loves Planting

题目：CF444E DZY Loves Planting

和【宏律】有一点像，第一眼是二分+带权二分图匹配，实际上不需要。。

可以把所有边按照边权排序然后搞一搞，具体做法就是每次合并两个端点，然后把子树大小和权值也合并。直到某个子树的大小已经超过剩余权值，这条边就是我们需要求的答案。

#include<stdio.h>
#include<algorithm>
#define it register int
#define il inline
using namespace std;
const int N=1000005;
int h[N],nxt[N],adj[N],w[N],fa[N],cnt[N],n,tot;
struct ky{
    int l,r,num;
    bool operator<(const ky&p)const{
        return num<p.num;
    }
}a[N];
il int fd(it x){
    return fa[x]==x?x:fa[x]=fd(fa[x]);
}
il void fr(int &num){
    num=0;char c=getchar();int p=1;
    while(c<'0'||c>'9') c=='-'?p=-1,c=getchar():c=getchar();
    while(c>='0'&&c<='9') num=num*10+c-'0',c=getchar();
    num*=p;
}   
int main(){ 
    fr(n);
    for(it i=1;i<n;++i) fr(a[i].l),fr(a[i].r),fr(a[i].num);
    sort(a+1,a+n);
    for(it i=1;i<=n;++i) fr(w[i]),fa[i]=i,cnt[i]=1,tot+=w[i];
    for(it i=1,t1,t2;i<n;++i){
        t1=fd(a[i].l),t2=fd(a[i].r);
        fa[t1]=t2,cnt[t2]+=cnt[t1],w[t2]+=w[t1];
        if(cnt[t2]>tot-w[t2]) return printf("%d",a[i].num),0;
    }
    printf("%d",a[n-1].num);
    return 0;
}