最小生成树练习3(普里姆算法Prim)

风萧萧兮易水寒,壮士要去敲代码。本女子开学后再敲了。。00C62DE6

poj1258 Agri-Net(最小生成树)水题。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 using namespace std;
 5 const int inf=0x3f3f3f3f;
 6 const int N=101;
 7 int n,m;
 8 int g[N][N],low[N];
 9 void prim(int u0){
10     int i,j,mi,v,ans=0;
11     for(i=0;i<n;++i){
12         low[i]=g[u0][i];
13     }
14     low[u0]=-1;
15     for(i=1;i<n;++i){
16         mi=inf;
17         v=-1;
18         for(j=0;j<n;++j)
19             if(low[j]!=-1&&low[j]<mi){
20                 v=j; mi=low[j];
21             }
22         ans+=low[v];
23         low[v]=-1;
24         for(j=0;j<n;++j){
25             if(g[v][j]<low[j]){
26                 low[j]=g[v][j];
27             }
28         }
29     }
30     printf("%d
",ans);
31 }
32 int main(){
33     int t,i,j;
34     while(scanf("%d",&n)==1){
35         memset(g,inf,sizeof(inf));
36         for(i=0;i<n;++i)
37             for(j=0;j<n;++j)
38             scanf("%d",&g[i][j]);
39         prim(0);
40     }
41     return 0;
42 }
View Code

poj1751 Highways(最小生成树)水题。已修好的路赋值为0。不用sqrt计算距离,因为题目没要求。做完这题后把自学时的模板改了一点,因为一开始的模板太挫。。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #include<cmath>
 5 using namespace std;
 6 const int inf=0x3f3f3f3f;
 7 const int N=751;
 8 int n,m;
 9 int g[N][N],low[N],near[N],x[N],y[N];
10 void prim(int u0){
11     int i,j,mi,v;
12     for(i=1;i<=n;++i){
13         low[i]=g[u0][i];
14         near[i]=u0;
15     }
16     low[u0]=-1;
17     for(i=1;i<n;++i){
18         mi=inf;
19         v=-1;
20         for(j=1;j<=n;++j)
21             if(low[j]!=-1&&low[j]<mi){
22                 v=j; mi=low[j];
23             }
24         if(mi!=0)printf("%d %d
",near[v],v);
25         low[v]=-1;
26         for(j=1;j<=n;++j){
27             if(g[v][j]<low[j]){
28                 low[j]=g[v][j];
29                 near[j]=v;
30             }
31         }
32     }
33 }
34 int main(){
35     int t,i,j;
36     scanf("%d",&n);
37     memset(g,inf,sizeof(g));
38     for(i=1;i<=n;++i) scanf("%d%d",&x[i],&y[i]);
39     for(i=1;i<=n;++i){
40         for(j=i+1;j<=n;++j){
41             g[i][j]=g[j][i]=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
42         }
43     }
44     scanf("%d",&m);
45     while(m--){
46         scanf("%d%d",&i,&j);
47         g[i][j]=g[j][i]=0;
48     }
49     prim(1);
50     return 0;
51 }
View Code

poj1789 Truck History(最小生成树)每种卡车类型都是由其他卡车类型派生出来,第一种除外。理解:顶点为卡车类型,边的权值为类型对(t0,d0)的距离(编码中不同字符的位置数目)。。看懂题看得头疼,其实不难做。

 1 #include<cstdio>
 2 #include<cstring>
 3 using namespace std;
 4 const int inf=0x3f3f3f3f;
 5 const int N=2001;
 6 int n;
 7 char code[N][8];
 8 int g[N][N],low[N],a[N],ans;
 9 void prim(int u0){
10     int i,j,mi,v;
11     ans=0;
12     for(i=0;i<n;++i){
13         low[i]=g[u0][i];
14     }
15     low[u0]=-1;
16     for(i=1;i<n;++i){
17         mi=inf;
18         v=-1;
19         for(j=0;j<n;++j)
20             if(low[j]!=-1&&low[j]<mi){
21                 v=j; mi=low[j];
22             }
23         ans+=low[v];
24         low[v]=-1;
25         for(j=0;j<n;++j){
26             if(g[v][j]<low[j])
27                 low[j]=g[v][j];
28         }
29     }
30     printf("The highest possible quality is 1/%d.
",ans);
31 }
32 int main(){
33     int i,j,k,d;
34     while(scanf("%d",&n),n){
35         memset(g,inf,sizeof(g));
36         for(i=0;i<n;++i) scanf("%s",code[i]);
37         for(i=0;i<n-1;++i){
38             for(j=i+1;j<n;++j){
39                 d=0;
40                 for(k=0;k<7;++k)
41                     d+=(code[i][k]!=code[j][k]);
42                 g[i][j]=g[j][i]=d;
43             }
44         }
45         prim(0);
46     }
47     return 0;
48 }
View Code

poj2349 Arctic Network(最小生成树)求出最小生成树,将边从大到小排序,前s-1条用s个卫星通信,第s大边即为答案。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #include<cmath>
 5 using namespace std;
 6 const int inf=0x3f3f3f3f;
 7 const int N=501;
 8 int n,s;
 9 double g[N][N],low[N],d[N],x[N],y[N];
10 int cmp(int a,int b){
11     return a > b;
12 }
13 void prim(int u0){
14     int i,j,mi,v;
15     int cnt=0;
16     for(i=0;i<n;++i){
17         low[i]=g[u0][i];
18     }
19     low[u0]=-1;
20     for(i=1;i<n;++i){
21         mi=inf;
22         v=-1;
23         for(j=0;j<n;++j)
24             if(low[j]!=-1&&low[j]<mi){
25                 v=j; mi=low[j];
26             }
27         d[cnt++]=low[v];
28         low[v]=-1;
29         for(j=0;j<n;++j){
30             if(g[v][j]<low[j])
31                 low[j]=g[v][j];
32         }
33     }
34     sort(d,d+cnt,cmp);
35     printf("%.2f
",d[s-1]);
36 }
37 int main(){
38     int t,i,j;
39     double d;
40     scanf("%d",&t);
41     while(t--){
42         memset(g,inf,sizeof(g));
43         scanf("%d%d",&s,&n);
44         for(i=0;i<n;++i) scanf("%lf%lf",&x[i],&y[i]);
45         for(i=0;i<n;++i){
46             for(j=i+1;j<n;++j){
47                 d=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
48                 g[i][j]=g[j][i]=d;
49             }
50         }
51         prim(0);
52     }
53     return 0;
54 }
View Code

poj3026 Borg Maze(最小生成树,bfs预处理)因为这题我几天没打码了,今天做出来感觉也不是很难。还是深感基础之弱内心痛苦。。对了,记得看discuss哦,如果你不想多挥霍青春的话。。。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #include<queue>
 5 using namespace std;
 6 const int inf=0x3f3f3f3f;
 7 const int N=51;
 8 const int M=102;
 9 int x,y;
10 int n;
11 int dir[][2]={0,1,0,-1,1,0,-1,0};
12 int g[M][M],low[M];
13 char mp[N][N];
14 int G[N][N];
15 int vd[N][N];
16 void bfs(int sx,int sy){
17     queue<pair<int,int> >q;
18     while(!q.empty())q.pop();
19     memset(vd,-1,sizeof(vd));
20     vd[sx][sy]=0;
21     q.push(make_pair(sx,sy));
22     while(!q.empty()){
23         pair<int,int>u=q.front(); q.pop();
24         if(G[u.first][u.second]>0)
25             g[G[sx][sy]][G[u.first][u.second]]=vd[u.first][u.second];
26         for(int i=0;i<4;++i){
27             int xx=u.first+dir[i][0];
28             int yy=u.second+dir[i][1];
29             if(mp[xx][yy]=='#'||vd[xx][yy]!=-1)continue;
30                 vd[xx][yy]=vd[u.first][u.second]+1;
31                 q.push(make_pair(xx,yy));
32         }
33     }
34 }
35 void prim(int u0){
36     int i,j,mi,v,ans=0;
37     for(i=1;i<n;++i) low[i]=g[u0][i];
38     low[u0]=-1;
39     for(i=1;i<n-1;++i){
40         mi=inf;
41         v=-1;
42         for(j=1;j<n;++j)
43             if(low[j]!=-1&&low[j]<mi){
44                 v=j; mi=low[j];
45             }
46         ans+=low[v];
47         low[v]=-1;
48         for(j=1;j<n;++j){
49             if(g[v][j]<low[j]){
50                 low[j]=g[v][j];
51             }
52         }
53     }
54     printf("%d
",ans);
55 }
56 int main(){
57     int t,i,j;
58     scanf("%d",&t);
59     while(t--){
60         scanf("%d%d",&x,&y);
61         gets(mp[0]);
62         for(i=0;i<y;++i) gets(mp[i]);
63         n=1;
64         for(i=0;i<y;++i){
65             for(j=0;j<x;++j){
66                 if(mp[i][j]=='A'||mp[i][j]=='S')
67                     G[i][j]=n++;
68                 else G[i][j]=-1;
69             }
70         }
71         for(i=0;i<y;++i)
72             for(j=0;j<x;++j)
73             if(G[i][j]>0) bfs(i,j);
74         prim(1);
75     }
76     return 0;
77 }
View Code
原文地址:https://www.cnblogs.com/GraceSkyer/p/5790215.html