标签：const int 边权 Codeforces Queries 582 include 询问 define

题目链接

题意：给出一棵树，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;
}

标签：const,int,边权,Codeforces,Queries,582,include,询问,define
来源： https://blog.csdn.net/qq_41292370/article/details/100525056

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。