1 kruskal
2 #include <vector>
3 #include <iostream>
4 #include <cstdio>
5 #include <cstring>
6 #include <algorithm>
7 #define INF 0x3f3f3f3f
8
9 using namespace std;
10 int n,m;
11 struct data
12 {
13 int u,v,w;
14 bool vis;
15 } p[20010];
16 vector<int>G[110];
17 int per[110],maxd[110][110];
18 bool cmp(data a,data b)
19 {
20 return a.w < b.w;
21 }
22 int Union_Find(int x)
23 {
24 return x == per[x] ? x: per[x] = Union_Find(per[x]);
25 }
26 void kruskal()
27 {
28 sort(p,p+m,cmp);
29 for(int i=0; i<=n; i++)//初始化
30 {
31 G[i].clear();
32 G[i].push_back(i);
33 per[i]=i;
34 }
35 int sum=0,k=0;//sum是最小生成树的值
36 for(int i=0; i<m; i++){
37 if(k==n-1) break;
38 int x1=Union_Find(p[i].u),x2=Union_Find(p[i].v);
39 if(x1!=x2){
40 k++;
41 p[i].vis=1;//这条边已经用过了
42 sum+=p[i].w;
43 int len_x1=G[x1].size();
44 int len_x2=G[x2].size();
45 for(int j=0; j<len_x1; j++) //更新两点之间距离的最大值
46 for(int k=0; k<len_x2; k++)
47 maxd[G[x1][j]][G[x2][k]]=maxd[G[x2][k]][G[x1][j]]=p[i].w;//因为后面的边会越来越大，所以这里可以直接等于当前边的长度
48 per[x1]=x2;
49 //因为per[x1] = x2,在Union_Find函数中要寻找和x1相关联节点的跟节点的时候，都会找到x2，所以这里不用再去更新和x1节点相连的节点
50 //十分感谢Self-Discipline博主，提出此问题
51 // int tem[110];
52 // for(int j=0; j<len_x2; j++)//现在已经属于一棵树了，那么我们就将点添加到相应的集合中
53 // tem[j]=G[x2][j];
54 for(int j=0; j<len_x1; j++)
55 G[x2].push_back(G[x1][j]);
56 // for(int j=0; j<len_x2; j++)
57 // G[x1].push_back(tem[j]);
58 }
59 }
60 int cisum=INF;//次小生成树的权值
61 for(int i=0; i<m; i++)
62 if(!p[i].vis)
63 cisum=min(cisum,sum+p[i].w-maxd[p[i].u][p[i].v]);
64 if(cisum>sum)
65 printf("%d\n",sum);
66 else
67 printf("Not Unique!\n");
68 }
69 int main()
70 {
71 int T;
72 scanf("%d\n",&T);
73 while(T--){
74 scanf("%d%d",&n,&m);
75 for(int i=0; i<m; i++){
76 scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w);
77 p[i].vis = false;
78 }
79 kruskal();
80 }
81 return 0;
82 }
83
84 prim
85 #include<iostream>
86 #include<string>
87 #include<cstdio>
88 #include<map>
89 #include<cstring>
90 #include<cmath>
91 #include<algorithm>
92 using namespace std;
93 const int INF=99999999;
94 const int N=1010;
95 struct point
96 {
97 int x,y;
98 } p[N];
99 double G[N][N],dist[N];
100 double path[N][N];//从i到j的路径上最大边的权值
101 int population[N];//每个城市的人口数
102 int pre[N],visit[N];
103 bool used[N][N];//边是否在该MST中
104 int n;
105 inline double Dist(point v1,point v2)
106 {
107 return sqrt(double(v1.x-v2.x)*(v1.x-v2.x)+double(v1.y-v2.y)*(v1.y-v2.y));
108 }
109 double Prim()
110 {
111 double Mst=0;
112 memset(visit,0,sizeof(visit));
113 memset(used,0,sizeof(used));
114 memset(path,0,sizeof(path));
115 visit[1]=1;
116 for(int i=1; i<=n; ++i){
117 dist[i] = G[1][i];
118 pre[i] = 1;
119 }
120 for(int i=1; i<n; ++i){
121 int u=-1;
122 for(int j=1; j<=n; ++j){
123 if(!visit[j]){
124 if(u==-1||dist[j]<dist[u])
125 u=j;
126 }
127 }
128 used[u][pre[u]]=used[pre[u]][u] = true;//加入MST
129 Mst+=G[pre[u]][u];
130 visit[u]=1;
131 for(int j=1; j<=n; ++j){
132 if(visit[j]&&j!=u)//求从u到j的路径上最大边的权值
133 {
134 path[u][j]=path[j][u]=max(path[j][pre[u]],dist[u]);
135 //printf("path[j][pre[u]]:%lf\n",path[j][pre[u]]);
136 }
137 if(!visit[j]){
138 if(dist[j]>G[u][j])//更新相邻顶点的dist
139 {
140 dist[j]=G[u][j];
141 pre[j]=u;
142 }
143 }
144 }
145 }
146 return Mst;
147 }
148 int main()
149 {
150 int tcase;
151 scanf("%d",&tcase);
152 while(tcase--){
153 scanf("%d",&n);
154 memset(G,0,sizeof(G));
155 for(int i=1; i<=n; ++i)
156 scanf("%d%d%d",&p[i].x,&p[i].y,&population[i]);
157 for(int i=1; i<=n; ++i){
158 for(int j=1; j<=n; ++j){
159 if(i!=j)
160 G[i][j]=Dist(p[i],p[j]);
161 }
162 }
163 double Mst=Prim();
164 double res=-1;
165 for(int i=1; i<=n; ++i){
166 for(int j=1; j<=n; ++j)
167 if(i!=j){
168 if(used[i][j])
169 res=max(res,(population[i]+population[j])/(Mst-G[i][j]));
170 else
171 res=max(res,(population[i]+population[j])/(Mst-path[i][j]));
172 }
173 }
174 printf("%.2f\n",res);
175 }
176 return 0;
177 }
原文地址：https://www.cnblogs.com/pangbi/p/11614445.html
时间： 2024-10-10 09:04:59