HDU 1102 Constructing Roads

最新推荐文章于 2024-07-21 14:30:00 发布

原创最新推荐文章于 2024-07-21 14:30:00 发布 · 206 阅读

0 ·

CC 4.0 BY-SA版权

图论同时被 3 个专栏收录

67 篇文章

订阅专栏

HDOJ

22 篇文章

订阅专栏

图论 - 最小生成树、并查集

21 篇文章

订阅专栏

本文介绍了一个基于Prim算法实现最小生成树的C++程序实例。该程序能够处理顶点间的边权重，并通过移除特定边来求解最小生成树问题。适用于解决网络设计等实际问题。

传送门

HDU 3371的弱化版，怎么都能过。

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;

const int MAXN = 1e2 + 3;
const int INF = 1e9;

int N, Q;
int g[MAXN][MAXN];
int d[MAXN];
bool vis[MAXN];
int ans;

void init()
{
	//fill(g[0], g[0] + MAXN*MAXN, INF);       // 没必要了，都直接邻接矩阵输入了
	fill(d + 1, d + N + 1, INF);
	fill(vis + 1, vis + N + 1, false);
	ans = 0;
}

int prim(int s)
{
	d[s] = 0;
	for (int i = 0; i < N; i++)
	{
		int mind = INF, u = -1;
		for (int j = 1; j <= N; j++)
		{
			if (!vis[j] && d[j] < mind)
			{
				mind = d[j];
				u = j;
			}
		}
		
		vis[u] = true;
		ans += mind;

		for (int j = 1; j <= N; j++)
		{
			if (!vis[j] && g[u][j] != INF)     // 这个!=INF的判断其实也没必要了
			{
				if (g[u][j] < d[j])
					d[j] = g[u][j];
			}
		}
	}
	return ans;
}

int main()
{
	int a, b;
	for (; ~scanf("%d", &N);) 
	{
		init();
		for (int i = 1; i <= N; i++)
		{
			for (int j = 1; j <= N; j++)
			{
				scanf("%d", &a);
				/*if (i != j)*/ g[i][j] = a;  // 这个判断不用加，因为prim算法里有!vis[j]限制着，自己不可能再访问自己   
			}
		}
		scanf("%d", &Q);
		for (int i = 0; i < Q; i++)           // n(n+1)/2 其实就是 n(n-1)/2 + n
		{
			scanf("%d%d", &a, &b);
			g[a][b] = g[b][a] = 0;            //
		}
		printf("%d\n", prim(1));
	}

    return 0;
}