题目大意:
输入N表示现存在的村庄数目,接下来输入N*N规模的邻接矩阵来表示村庄间道路的信息,之后输入Q表示目前已存在的道路数目,随着输入Q条已存在的道路。求出使得所有村庄畅通的最.小代价。
算法思想:
1.新建一个邻接矩阵存储图的信息。
2.接着,将已经存在的道路的代价置为0,因为这样使得在最小生成树中对已存在的道路不产生代价。
3.用一个Edge的结构体数组存储所有的边。
4.将所有的边按权值非递减的顺序排序。
5.更具克鲁斯卡尔算法产生最小生成树计算最小的花费代价。
代码如下:
#include <iostream> #include <cstring> #include <algorithm> using namespace std; const int MAXN=5500; int G[105][105];//存储图的邻接矩阵 /*边结构*/ typedef struct { int start; int end; int value; }Edge; Edge road[MAXN]; /*排序时用的比较函数*/ bool cmp(Edge a,Edge b){ if(a.value<b.value) return true; return false; } /*顶点表*/ int node[110]; /*并查集函数*/ int Find_set(int n){ if(node[n]==-1) return n; return node[n]=Find_set(node[n]); } /*合并两棵树*/ bool Merge(int a,int b){ int r1=Find_set(a); int r2=Find_set(b); if(r1==r2) return false; if(r1<r2) node[r2]=r1; if(r2<r1) node[r1]=r2; return true; } int Kruskal(int N,int M,int cost){ int num=0; for(int i=0;i<M;i++){ if(Merge(road[i].start,road[i].end)){ num++; cost+=road[i].value; } if(num==N-1) break; } return cost; } int main(){ int N,Q,k;//乡村数 已修道路数 总的道路数 int a,b;//标记已建道路的两个端点 long cost;//记录最小花费代价 while(cin>>N){ cost=0; k=0; memset(node,-1,sizeof(node)); for(int i=1;i<=N;i++)//输入邻接矩阵 for(int j=1;j<=N;j++) cin>>G[i][j]; cin>>Q; for(int i=1;i<=Q;i++){//输入已经建好的道路信息 cin>>a>>b; G[a][b]=0;//将修路代价置为0 G[b][a]=0; } for(int i=1;i<=N;i++)//读取所有的道路信息 for(int j=1;j<i;j++){ road[k].start=i; road[k].end=j; road[k].value=G[i][j]; k++; } sort(road,road+k,cmp);//对道路按边的权值进行排序 cost=Kruskal(N,k,cost);//生成最小生成树 cout<<cost<<endl; } return 0; }
时间: 2024-10-08 11:25:52