Codeforces Round #582 (Div. 3)-G. Path Queries-并查集

Codeforces Round #582 (Div. 3)-G. Path Queries-并查集

---

【Problem Description】

给你一棵树，求有多少条简单路径\((u,v)\)，满足\(u\)到\(v\)这条路径上的最大值不超过\(k\)。\(q\)次查询。

【Solution】

并查集

将所有边按权值从小到大排序，查询值\(k_i\)也从小到大排序。对于每次查询的值\(k_i\)，将边权小于等于\(k_i\)的边通过并查集合并在一起。对于合并后的联通块，每个联通块对答案的贡献为\(\frac{size\times(size-1)}{2}\)，所有联通块的贡献直接求和即可。所以每次合并两个节点时，先将这两个节点所在的联通块的贡献减去，再加上合并后的贡献即可。

---

【Code】

/*
 * @Author: Simon 
 * @Date: 2019-09-07 10:00:30 
 * @Last Modified by: Simon
 * @Last Modified time: 2019-09-07 10:29:11
 */
#include
using namespace std;
typedef int Int;
#define int long long
#define INF 0x3f3f3f3f
#define maxn 200005
struct node{
    int u,v,w;
    node(){}
    node(int u,int v,int w):u(u),v(v),w(w){}
    bool operator <(const node&a)const{
        return w>n>>m;init(n);
    for(int i=1;i>g[i].u>>g[i].v>>g[i].w;
    }
    sort(g+1,g+n);
    for(int i=1;i<=m;i++){
        cin>>q[i].w;q[i].u=i;
    }
    sort(q+1,q+m+1);
    int pos=1;
    for(int i=1;i<=m;i++){
        while(pos