SWUSTOJ #1075 最小生成树

模板题,不想多说,虽然题目给了一些代码,但是我觉得太麻烦了,各位直接看代码吧,我相信大家能够理解,实在理解不了,可以私聊我
在这里插入图片描述

#include <iostream>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <cstring>
#define int long long
#define INF 0X3f3f3f3f
#define fr(i,a,b) for(int i=(a);i<(b);i++)
using namespace std;
int dis[100][100];
int d[100];
int mp[100];
int flag[100];
map<char,int>mp1;
map<int,char>mp2;
int m;
int n;
void Prime(int v)
{
	//int sum;
	//sum = 0;
	for(int i = 1; i <= n; i++){
		d[i] = dis[v][i];
		mp[i] = v;
	}
	d[v] = 0;
	flag[v] = 1;
	int k,p=-1;
	for(int i = 1; i < n; i++)
	{
		int minn = INF;
		for(int j = 1; j <= n; j++)
		{
			if(minn > d[j] && flag[j] == 0)
			{
				k = j;
				minn = d[j];
			}
		}
		flag[k] = 1;
		if(k==p){
			return;
		}
		p=k;
		cout <<"("<<mp2[mp[k]]<<","<<mp2[k]<<")";
		//sum += d[k];
		for(int j =1; j <= n; j++)
		{
			if(flag[j] == 0 && d[j] > dis[k][j]){
				mp[j]= k;
				d[j] = dis[k][j];
			}
		}
	}
	//return sum;
}
signed main(){
	cin >> n >> m;
	fr(i,0,n){
		char A;
		cin >> A;
		mp1[A]=i+1;
		mp2[i+1]=A;
	}
	int t = m;
	memset(dis,INF,sizeof(dis));
	while(t--){
		char x,y;
		int w;
		cin >> x >> y >> w;
		dis[mp1[x]][mp1[y]]=w;
		dis[mp1[y]][mp1[x]]=w;
	}
	fr(i,0,n){
		if(!flag[i+1]){
			Prime(i+1);
		}
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
好的,我来续写代码: ``` # 导入必要的库 import networkx as nx import numpy as np # 读取表格数据 data = np.loadtxt("data.csv", delimiter=",", dtype=int) # 构建有向图 G = nx.DiGraph() for edge in data: G.add_edge(edge[1], edge[2], weight=edge[3]) # 计算最小生成树 UG = G.to_undirected() T = nx.minimum_spanning_tree(UG) # 计算每个节点对之间的最短路径 shortest_paths = dict(nx.shortest_path_length(T)) # 选择最优的5条额外边 best_edges = set() for _ in range(5): best_edge = None best_delta = np.inf for i in G.nodes(): for j in G.nodes(): if i != j and not G.has_edge(i, j): delta = 0 for k in G.nodes(): if k != i and k != j: delta += shortest_paths[k][i] + shortest_paths[k][j] - 2 * shortest_paths[k][T.nodes()[0]] # 计算新增边对路径长度的贡献 if delta < best_delta: best_edge = (i, j) best_delta = delta best_edges.add(best_edge) G.add_edge(best_edge[0], best_edge[1], weight=0) # 将新增边加入有向图中 # 计算修改后的最小生成树 UG = G.to_undirected() T = nx.minimum_spanning_tree(UG) # 输出结果 print("最小生成树中的节点:", T.nodes()) print("最小生成树中的边:", T.edges()) print("每个节点之间的最短路径:", shortest_paths) print("新增的5条边:", best_edges) ``` 上述代码中,我们首先构建有向图 `G`,然后计算其最小生成树 `T`,使用 `nx.shortest_path_length` 函数计算每个节点对之间的最短路径,并选择最优的5条额外边。在计算新增边对路径长度的贡献时,我们使用了最小生成树中任意一个节点作为参考点,这里选择了 `T.nodes()[0]`。最后,将新增边加入有向图中,重新计算其最小生成树,输出结果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值