Codeforces Round #582 (Div. 3) G. Path Queries（树上分块）

题目链接

题意：给出一棵树，n个节点，n-1条边，每条边有自己的权值，现在有m次询问，每次询问给出一个值，现在要求出有多少对(u,v)节点（u,v节点满足之间的简单路径中的边权最大值不超过给定询问值）。

题解：首先这个题我们可以先将询问离线，我们先将树上的边权按从小到大排序，排序了过后我们就能够保证当前加入的边权一定是在已加入的边权中最大的。然后我们用并查集来合并，每次合并我们将加入这条边的贡献加到对应边权上，注意，要是每次都加我们肯定会加多，所以我们这里要判断是否加重，如果当前两个端点在不同连通块中，说明这时候我们就要加上这条边带来的贡献，要是这两个点在一个连通块中，则之前就加过两点的贡献，每次连上这条边对点对数量的贡献就是sum[a]∗sum[b]。

#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<cmath>
#include<stack>
#include<string>

const int mod = 998244353;
const int maxn = 1e6 + 5;
const int inf = 1e7;
const long long onf = 1e18;
#define me(a, b) memset(a,b,sizeof(a))
#define lowbit(x) x&(-x)
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI 3.14159265358979323846
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int n, m;
int q[maxn], root[maxn], sum[maxn];
ll ans[maxn];

struct node {
    int u, v, val;

    bool friend operator<(node a, node b) {
        return a.val < b.val;
    }
} a[maxn];

int Find(int x) {
    if (x == root[x])
        return x;
    return root[x] = Find(root[x]);
}

void init() {
    for (int i = 1; i <= n; i++) {
        sum[i] = 1, root[i] = i;
    }
}

int main() {
    //freopen("1.in", "r", stdin);
    scanf("%d%d", &n, &m);
    init();
    for (int i = 1; i < n; i++)
        scanf("%d%d%d", &a[i].u, &a[i].v, &a[i].val);
    sort(a + 1, a + n);
    int End = 0;
    for (int i = 1; i <= m; i++) {
        scanf("%d", &q[i]);
        End = max(End, q[i]);
    }
    for (int i = 1; i < n; i++) {
        int x1 = Find(a[i].u);
        int y1 = Find(a[i].v);
        if (x1 != y1) {
            ll temp = (ll) sum[x1] * sum[y1];
            sum[y1] += sum[x1], sum[x1] = 0;///将两个连通块合并成一个
            root[x1] = y1;
            ans[a[i].val] += temp;
        }
    }
    for (int i = 1; i <= End; i++)///这点很好理解，因为是求满足小于给定边权的点的对数。
        ans[i] += ans[i - 1];
    for (int i = 1; i <= m; i++)
        printf("%lld ", ans[q[i]]);
    return 0;
}