hdu 3938 Portal

最新推荐文章于 2026-04-15 10:02:42 发布

原创最新推荐文章于 2026-04-15 10:02:42 发布 · 678 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

并查集

最小生成树

本文介绍了一种离线算法，用于解决给定无向图中路径花费小于特定值的问题。通过最小生成树Kruskal算法，筛选出满足条件的路径，并通过并查集实现路径计数。

题意描述

简单的讲就是，给你一张无向图，求有多少条路径使得路径上的花费小于L，这里路径上的花费是这样规定的，a、b两点之间的多条路径中的最长的边最小值！

题目分析

因为考虑到数据的大小所以需采用离线算法先计算出结果后在统一输出结果。这里首先要从a到b的所有路径中筛选出这样一条路径（路径中的最长边是所有路径中最小），那么这就需要采用最小生成树kruskal的思想，那么任意两个集合A,B之间可以形成的路径即为num[A]*num[B]。

AC代码

#include <queue>
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 10010;
const int M = 50010;
const int Q = 10010;

int n, m, k;
int father[N], sum[N];
long long num[Q];

struct edge{
	int u, v, c;
}node[M];

struct query{
	int l, id;
}q[Q];

bool cmp1(const edge &A, const edge &B){
	return A.c < B.c;
}

bool cmp2(const query &A, const query &B){
	return A.l < B.l;
}

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

int find(int x){
	if(x != father[x])
		father[x] = find(father[x]);
	return father[x];
}

int Union(int x, int y){
	int xx = find(x), yy = find(y);
	if(xx == yy) return 0;

	father[yy] = xx;
	int tmp = sum[yy] * sum[xx];
	sum[xx] += sum[yy];
	return tmp;
}

int main(){
	while(scanf("%d%d%d",&n, &m, &k) == 3){
		for(int i=0; i<m; i++)
			scanf("%d%d%d",&node[i].u, &node[i].v, &node[i].c);
		sort(node, node+m, cmp1);
		for(int i=0; i<k; i++)
			scanf("%d",&q[i].l), q[i].id = i;
		sort(q, q+k, cmp2);
		init();
		int cnt = 0;
		long long ans = 0;
		for(int i=0; i<k; i++){
			while(node[cnt].c <= q[i].l && cnt < m){
				ans += Union(node[cnt].u, node[cnt].v);
				cnt++;
			}
			num[q[i].id] = ans;
		}
		for(int i=0; i<k; i++)
			printf("%lld\n", num[i]);
	}
	return 0;
}