pku 1751

1. #include <stdio.h>
2. #define INF 200000000
3. #define MAX 751
4. int path[MAX];
5. int matrix[MAX][MAX];
6. int main()
7. {
8.     int x[MAX],y[MAX];
10.     int N,M;
11.     int p1,p2;
12.     int temp1,temp2;
13.     int i,j;
15.     int cost[MAX];
16.     int min;
17.     int u;
19.     scanf("%d",&N);
20.     for(i=1;i<=N;i++)
21.         scanf("%d%d",&x[i],&y[i]);
23.     for(i=1;i<=N;i++)
24.         for(j=i;j<=N;j++)
25.             if(i==j)
26.                 matrix[i][i]=INF;
27.             else
28.             {
29.                 temp1=x[i]-x[j];
30.                 temp2=y[i]-y[j];
31.                 matrix[j][i]=matrix[i][j]=temp1*temp1+temp2*temp2;
32.             }
34.     scanf("%d",&M);
35.     for(i=1;i<=M;i++)
36.     {
37.         scanf("%d%d",&p1,&p2);
38.         matrix[p1][p2]=matrix[p2][p1]=0;
39.     }
40.     
41.     for(i=1;i<=N;i++)
42.     {
43.         cost[i]=matrix[1][i];
44.         path[i]=1;
45.     }
46.     cost[1]=-1;
48.     for(i=1;i<=N-1;i++)
49.     {
50.         min=INF;
51.         for(j=1;j<=N;j++)
52.             if(cost[j]!=-1 && cost[j]<min)
53.             {
54.                 u=j;
55.                 min=cost[j];
56.             }
57.         if(min>0)
58.             printf("%d %d/n",path[u],u);
59.         cost[u]=-1;
60.         for(j=1;j<=N;j++)
61.             if(cost[j]!=-1 && cost[j]>matrix[u][j])
62.             {
63.                 path[j]=u;
64.                 cost[j]=matrix[u][j];
65.             }
66.     }
67.     return 0;
68. }
69. /*18 4543344(5) xiaofengsheng 2404K 79MS C++ 1028B 2009-01-09 15:38:56
70. 当边稠密时最好用prim,而不宜用Kruskal
71. (这里是已有添加好边的minimum spanning tree problem)
72. 经过几次提交发现：
73.     1.typedef struct point
74.     {
75.         int x;
76.         int y;
77.     }point;
78.     用结构体稍微比数组 int x[MAX],y[MAX]; 慢,不过结构体结构清晰;
79.     2.if(i==j)
80.             matrix[i][i]=INF;
81.     开始时处理的不好，
82.     for(i=1;i<=N;i++)
83.         for(j=1;j<=N;j++)
84.             matrix[i][j]=INF;
86. 另外,多亏suikay代码的有益启发!thanks...
87. */
    95. #include <stdio.h>
    96. #define INF 200000000
    97. #define MAX 751
    98. typedef struct point
    99. {
    100.     int x;
    101.     int y;
    102. }point;
    103. point p[MAX];
    104. int path[MAX];
    105. int matrix[MAX][MAX];
    107. void main()
    108. {
    109.     int N,M;
    110.     int p1,p2;
    111.     int temp1,temp2;
    112.     int i,j;
    114.     int cost[MAX];
    115.     int min;
    116.     int u;
    118.     scanf("%d",&N);
    119.     for(i=1;i<=MAX;i++)
    120.         for(j=1;j<=MAX;j++)
    121.             matrix[i][j]=INF;
    123.     for(i=1;i<=N;i++)
    124.         scanf("%d%d",&p[i].x,&p[i].y);
    126.     for(i=1;i<=N;i++)
    127.         for(j=i+1;j<=N;j++)
    128.         {
    129.             temp1=p[i].x-p[j].x;
    130.             temp2=p[i].y-p[j].y;
    131.             matrix[j][i]=matrix[i][j]=temp1*temp1+temp2*temp2;
    132.         }
    134.     scanf("%d",&M);
    135.     for(i=1;i<=M;i++)
    136.     {
    137.         scanf("%d%d",&p1,&p2);
    138.         matrix[p1][p2]=matrix[p2][p1]=0;
    139.     }
    140.     
    141.     //prim
    142.     for(i=1;i<=N;i++)
    143.     {
    144.         cost[i]=matrix[1][i];
    145.         path[i]=1;
    146.     }
    147.     cost[1]=-1;
    149.     for(i=1;i<=N-1;i++)
    150.     {
    151.         min=INF;
    152.         for(j=1;j<=N;j++)
    153.             if(cost[j]!=-1 && cost[j]<min)
    154.             {
    155.                 u=j;
    156.                 min=cost[j];
    157.             }
    158.         if(min>0)
    159.             printf("%d %d/n",path[u],u);
    160.         cost[u]=-1;
    161.         for(j=1;j<=N;j++)
    162.             if(cost[j]!=-1 && cost[j]>matrix[u][j])
    163.             {
    164.                 path[j]=u;
    165.                 cost[j]=matrix[u][j];
    166.             }
    167.     }
    168. }
    169. //最开始的code