信息学奥赛一本通 1349:【例4-10】最优布线问题 | 1350:【例4-11】最短网络(agrinet)

【题目链接】

ybt 1349:【例4-10】最优布线问题
ybt 1350:【例4-11】最短网络(agrinet)

【题目考点】

1. 图论:最小生成树

记图中顶点数为V,边数为E

  • Prim算法
    复杂度: O ( V 2 ) O(V^2) O(V2)
  • Prim算法堆优化
    复杂度: O ( E l o g E ) O(E log E) O(ElogE)
  • Kruskal算法
    复杂度: O ( E l o g E ) O(E log E) O(ElogE)

【解题思路】

最小生成树模板题,求最小生成树所有边权加和。
该题输入的是邻接矩阵,因此使用邻接矩阵解决该问题。当然也可以保存为邻接表。

【题解代码】

解法1:Prim算法
  • 写法1:使用邻接矩阵
#include<bits/stdc++.h>
using namespace std;
#define N 105
int n, m, sum, dis[N], edge[N][N];//sum:最小生成树边权加和 edge:邻接矩阵 dis[i]:顶点i到已经确定的顶点中的最小距离 
bool vis[N];//vis[i]:顶点i是否已经被确定 
void prim()
{
	memset(dis, 0x3f, sizeof(dis));
	dis[1] = 0;
	for(int k = 1; k <= n; ++k)
	{
		int u = 0;
		for(int i = 1; i <= n; ++i)
			if(vis[i] == false && (u == 0 || dis[i] < dis[u]))
				u = i;
		vis[u] = true;
		sum += dis[u];//顶点u到已经确定的顶点的最短的边的长度是dis[u],把这条边加入最小生成树,最小生成树的边权值增加dis[u] 
		for(int v = 1; v <= n; ++v)
			if(edge[u][v] && vis[v] == false && dis[v] > edge[u][v])
				dis[v] = edge[u][v];
	}
}
int main()
{
	cin >> n;
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
			cin >> edge[i][j];
	prim();
	cout << sum;
	return 0;
}
  • 写法2:使用邻接表
#include<bits/stdc++.h>
using namespace std;
#define N 105
struct Edge
{
	int v, w;//v:到达顶点 w:边权值 
	Edge(){}
	Edge(int a, int b):v(a), w(b){}
};
vector<Edge> edge[N];//edge[i]:顶点i的所有邻接点 
int n, m, sum, dis[N];//sum:最小生成树边权加和 dis[i]:顶点i到已经确定的顶点中的最小距离 
bool vis[N];//vis[i]:顶点i是否已经被确定 
void prim()
{
	memset(dis, 0x3f, sizeof(dis));
	dis[1] = 0;
	for(int k = 1; k <= n; ++k)
	{
		int u = 0;
		for(int i = 1; i <= n; ++i)
			if(vis[i] == false && (u == 0 || dis[i] < dis[u]))
				u = i;
		vis[u] = true;
		sum += dis[u];//顶点u到已经确定的顶点的最短的边的长度是dis[u],把这条边加入最小生成树,最小生成树的边权值增加dis[u] 
		for(Edge e : edge[u])
		{
			int v = e.v, w = e.w;
			if(vis[v] == false && dis[v] > w)
				dis[v] = w;
		}
	}
}
int main()
{
	int w;
	cin >> n;
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
		{
			cin >> w;
			if(w)//如果权值w不为0 
				edge[i].push_back(Edge(j, w));
		}
	prim();
	cout << sum;
	return 0;
}
解法2:Prim算法堆优化 使用邻接矩阵
#include<bits/stdc++.h>
using namespace std;
#define N 105
struct Pair
{
	int v, d;//v:顶点 d:顶点v到已确定顶点的距离
	Pair(){}
	Pair(int a, int b):v(a), d(b){}
	bool operator < (const Pair &b) const
	{
		return b.d < d;//到已确定顶点距离更小的数对更优先 
	}
};
int n, m, sum, dis[N], edge[N][N];//sum:最小生成树边权加和 edge:邻接矩阵 dis[i]:顶点i到已经确定的顶点中的最小距离 
bool vis[N];//vis[i]:顶点i是否已经被确定
void prim()
{
	priority_queue<Pair> pq;
	memset(dis, 0x3f, sizeof(dis));
	dis[1] = 0;
	pq.push(Pair(1, 0));
	int visNum = 0;//已经访问的顶点数 
	while(pq.empty() == false)
	{
		int u = pq.top().v, d = pq.top().d;
		pq.pop();
		if(vis[u])
			continue;
		vis[u] = true;
		sum += d;
		if(++visNum == n)
			break;
		for(int i = 1; i <= n; ++i)
		{
			if(edge[u][i] && vis[i] == false && dis[i] > edge[u][i])
			{
				dis[i] = edge[u][i];
				pq.push(Pair(i, dis[i]));
			}
		}
	}
}
int main()
{
	cin >> n;
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
			cin >> edge[i][j];
	prim();
	cout << sum;
	return 0;
}
解法3:Kruskal算法
#include<bits/stdc++.h>
using namespace std;
#define N 105
struct Edge
{
	int f, t, w;
	Edge(){}
	Edge(int a, int b, int c):f(a),t(b),w(c){}
};
int n, m, edgeNum, sum, fa[N];
bool vis[N];
vector<Edge> edges;
bool cmp(Edge a, Edge b)
{
	return a.w < b.w;
}
void initFa()
{
	for(int i = 1; i <= n; ++i)
		fa[i] = i;
}
int find(int x)
{
	if(x == fa[x])
		return x;
	else
		return fa[x] = find(fa[x]);
}
void merge(int x, int y)
{
	fa[find(x)] = find(y);
}
void kruskal()
{
	sort(edges.begin(), edges.end(), cmp);
	for(Edge e : edges)
	{
		if(find(e.f) != find(e.t))
		{
			sum += e.w;
			merge(e.f, e.t);
			if(++edgeNum == n-1)
				return;
		}
	}
}
int main()
{
	int e;
	cin >> n;
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
		{
			cin >> e;	
			if(e)
				edges.push_back(Edge(i, j, e));
		}
	initFa();
	kruskal();
	cout << sum;
	return 0;
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值