HDU_1875_prim

最新推荐文章于 2020-02-22 15:11:14 发布

1emerald

最新推荐文章于 2020-02-22 15:11:14 发布

阅读量116

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： prim

本文链接：https://blog.youkuaiyun.com/qq_43580151/article/details/96275070

算法专栏收录该内容

174 篇文章

订阅专栏

本文介绍了一个解决HDU1875问题的算法实现，通过Prim算法来寻找符合条件的最小生成树。具体步骤包括：删除不符合条件的边，使用Prim算法确定最小生成树是否存在，如果存在则输出最小生成树的总权重。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

HDU 1875

一开始把不符合条件的边删掉
判断是否能生成最小生成树

#include <iostream>
#include <string.h>
#include <math.h>

#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e3;

double e[maxn][maxn];
int vis[maxn];
double dist[maxn];
int n,m;

struct point {
	int x,y;
	point(){}
	point(int a,int b) {
		x=a; y=b;
	}
} points[maxn];

double Distance(point& a,point& b) {
	double dx = (a.x - b.x)*(a.x - b.x);
	double dy = (a.y - b.y)*(a.y - b.y);
	return sqrt(dx + dy);
}

int prim() {
	memset(vis, 0 ,sizeof vis);
	vis[1] = 1; //随便找一个点加入最小生成树中
	for (int i=1; i<=n; i++) dist[i] = e[1][i]; // 更新每个点到树的距离
	double sum = 0;//最短的路径长度
	for (int i=1; i<n; i++) { // 开始找n-1条边
		double minn = INF; // 距离树最短的边
		int point = INF; // 距离树最近的点
		for (int i=1; i<=n; i++) {
			if (!vis[i] && dist[i]<minn ) {
				point = i;
				minn = dist[i];
			}
		}
		if (minn == INF) {
			printf("oh!\n");
			return 0;
		}
		vis[point] = true; // 该点访问过（在树中）
		sum += dist[point];
		for (int i=1; i<=n; i++) {
			if ( !vis[i] && e[point][i]<dist[i] )
				dist[i] = e[point][i];
		}
	}
	printf("%.1lf\n",sum*100);
	return 0;
}
int main() {
//	freopen("a.txt","r",stdin);
	int times; scanf("%d",&times);
	while (times--) {
		scanf("%d",&n);
		for (int i=1; i<=n; i++) {
			int x,y;
			scanf("%d%d",&x,&y);
			points[i] = point(x,y);
		}
		for (int i=1; i<=n; i++) 
			for (int j=1; j<=n; j++) {
				double dis = Distance(points[i],points[j]);
				if (10.00000<=dis && dis <=1000.000001)
					e[i][j] = e[j][i] = dis;
				else e[i][j] = (double)0x3f3f3f3f;
			}
		prim();
	}
	return 0;
}