POJ 1258 Agri-Net【最小生成樹(prim)】
今天brent講座,我竟然忘了!!!郁悶的要死,幸好是去上自習了,不然會更后悔。。。今天他講的生成樹,這個我還會一點,多多少少能自我安慰一下,不過在看課件的時候還是有所收獲,發現在實現最小生成樹的時候可以有兩種方法,差別特別小,但時間還是差一點的,不幸的是我原來的方法正是那個稍微慢一點的。簡單來說,有三個差別:
1):在初始化dis數組時一個初始化為map[s][i],(包括dis[s],其中s為源點),另一個初始化為無窮大,dis[s]=0;
2):一個將將flag[s]標記為true,另一個標記為false
3):最重要的一點,一個循環n-1次,一個循環n次
我把兩個算法貼出來對比一下:
16ms的:
0ms的:
以后要學會用這個快的。。。~
1):在初始化dis數組時一個初始化為map[s][i],(包括dis[s],其中s為源點),另一個初始化為無窮大,dis[s]=0;
2):一個將將flag[s]標記為true,另一個標記為false
3):最重要的一點,一個循環n-1次,一個循環n次
我把兩個算法貼出來對比一下:
16ms的:
1#include<stdio.h>
2#include<string.h>
3#include<stdlib.h>
4#define INF 0x1f1f1f1f
5#define M 102
6int map[M][M],dis[M];
7bool flag[M];
8int prim(int s,int n)
{
9int i,j,md,temp,total=0;
10memset(flag,false,sizeof(flag));
11flag[s]=true;
12for(i=1;i<=n;i++)
13dis[i]=map[s][i];
14for(i=1;i<n;i++)
{
15md=INF;
16for(j=1;j<=n;j++)
{
17if(!flag[j]&&dis[j]<md)
{
18temp=j;
19md=dis[j];
20}
21}
22total+=md;
23flag[temp]=true;
24for(j=1;j<=n;j++)
25if(!flag[j]&&map[temp][j]<dis[j])
26dis[j]=map[temp][j];
27}
28return total;
29}
30int main()
31{
32int i,j,k,n,cas;
33while(scanf("%d",&n)!=EOF)
{
34for(i=1;i<=n;i++)
35for(j=1;j<=n;j++)
{
36scanf("%d",&map[i][j]);
37if(map[i][j]==0) map[i][j]=INF;
38}
39printf("%d\n",prim(1,n));
40//system("PAUSE");
41}
42}
0ms的:
1#include<stdio.h>
2#include<string.h>
3#include<stdlib.h>
4#define INF 0x1f1f1f1f
5#define M 102
6int map[M][M],dis[M];
7bool flag[M];
8int prim(int s,int n)
{
9int i,j,md,temp,total=0;
10memset(flag,false,sizeof(flag));
11memset(dis,0x1f,sizeof(dis));
12dis[s]=0;
13for(i=1;i<=n;i++)
{
14md=INF;
15for(j=1;j<=n;j++)
{
16if(!flag[j]&&dis[j]<md)
{
17temp=j;
18md=dis[j];
19}
20}
21total+=md;
22flag[temp]=true;
23for(j=1;j<=n;j++)
24if(!flag[j]&&map[temp][j]<dis[j])
25dis[j]=map[temp][j];
26}
27return total;
28}
29int main()
30{
31int i,j,k,n,cas;
32while(scanf("%d",&n)!=EOF)
{
33for(i=1;i<=n;i++)
34for(j=1;j<=n;j++)
35scanf("%d",&map[i][j]);
36printf("%d\n",prim(1,n));
37}
38}
39
posted on 2010-05-09 00:19 M.J 閱讀(334) 評論(0) 編輯 收藏 引用 所屬分類: 最短路,最小生成樹





}
}

