并查集 二分-Codeforces-1213G Path Queries

题目链接
题意：
有n个点，编号1-n
保证这n个点来自于一个树，给出n-1条边的连接方式
问有多少种对(u，v) u<v满足u v之间的路上 没有长度超过q的边
样例+题目解释：

例如第一组样例构成的树如图所示
询问分别为：
5 2 3 4 1
第一次询问：
（1，2）（1，3）（1，4）（1，5）（1，6）（1，7）
（2，3）（2，4）（2，5）（2，6）（2，7）
（3，4）（3，5）（3，6）（3，7）
（4，5）（4，6）（4，7）
（5，6）（5，7）
（6，7）
共21条路径满足条件
第二次询问：
（1，2）（1，4）（1，5）
（2，4）（2，5）
（3，6）
（4，5）
共7条路径满足条件
……（省略其他询问）
思路：
刚开始看到这道题发现难度居然还没有1213F的难度高 和1213E难度一样，就果断来先补这道题了。
结果一看题发现和E完全不是一个难度水平啊…可能我这方面比较菜…
然后去看了别人AC的代码，还让队友给讲了才会写（我太菜了）
就来好好补这道题了。。

题目中说了这n个点来自于一个树，所以不用考虑某两个点之间不相连的情况
首先把这n-1条边按weight从小到大排序
然后利用并查集找出第i条边满足条件时，总共的情况数ans[i]
还用第一组样例来解释吧：
把n条边按weight排完序以后为：
1 2 1
2 4 1
4 5 2
3 6 2
3 2 3
5 7 4
然后用getans函数来得到每个ans的值
getans函数：（这里是我当初最难看懂的…）
生成第一条边时，将1的根节点变成2，2的sum值变成2，代表这一堆里现在有两个点
（把2的根节点变成1也是没问题的，但是要注意是把1的sum值变成2，因为之后每次要用的都是根节点的sum值，注意要对应）
ans[1]=0+11
//ans[i]表示第i条边满足条件时的情况总数
因为边是按weight从小到大排好的，所以满足ans[i-1]的边肯定满足ans[i]
11的意思在下面稍微复杂一点的地方解释

生成第二条边时，新组成的堆里的 新组成的边的个数为sum[2]*sum[1]，还需要再加上之前满足的ans[i-1]条边，同样和上面一样，把根节点变成4

……（省略之后的生成）
q次询问之后要用二分查找答案
因为边都是按从小到大排好的，所以只需找到满足 k[i].weight≤q的i的最大即可，ans[i]即为对应的答案。
感觉写完博客以后自己是真的懂了。。。如果还有哪里不小心写错的话，欢迎指出hhh

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn=200005;
ll n,m,sum[maxn],f[maxn],ans[maxn],i,j;
struct hh
{
    ll a,b,w;
} k[maxn];
bool cmp(hh k1,hh k2)
{
    return k1.w<k2.w;
}
ll fin(ll x)
{
    if(f[x]==x)
        return x;
    return f[x]=fin(f[x]);
}
void getans(ll a,ll b)
{
    a=fin(a);
    b=fin(b);
    ans[i]=ans[i-1]+sum[a]*sum[b];
    sum[b]+=sum[a];
    f[a]=b;
}
int main()
{
    scanf("%lld%lld",&n,&m);
    for(i=1; i<=n-1; i++)
    {
        scanf("%lld%lld%lld",&k[i].a,&k[i].b,&k[i].w);
        f[i]=i;
        sum[i]=1;
    }
    f[n]=n,sum[n]=1;
    sort(k+1,k+n,cmp);
    for(i=1; i<=n-1; i++)
    {
        getans(k[i].a,k[i].b);
    }
    while(m--)
    {
        ll z;
        scanf("%lld",&z);
        ll l=0,r=n-1,mid,ans1=0;
        while(l<=r)
        {
            mid=(l+r)/2;
            if(k[mid].w<=z)
            {
                l=mid+1;
                ans1=max(ans1,mid);
            }
            else
                r=mid-1;
        }
        printf("%lld ",ans[ans1]);
    }
}