poj 1287 heap+prime 最小生成树

 题目比较简单，可以不使用heap优化，这里作为练习

1. #include <iostream>
2. #include <algorithm>
3. using namespace std;
5. const int N = 600, INF = 1000000000;
7. int dist[N], g[N][N], map[N], heap[3*N], del[N];
8. int n, size;
10. void heap_up(int root)
11. {
12.     int t = heap[root], fa = root>>1;
13.     while(root > 1 && dist[t] < dist[heap[fa]])
14.     {
15.         heap[root] = heap[fa];
16.         map[heap[root]] = root;
17.         root = fa;
18.         fa >>= 1;
19.     }
20.     heap[root] = t;
21.     map[heap[root]] = root;
22. }
24. void heap_down(int root)
25. {
26.     int t = heap[root], ch = root<<1;
27.     while(ch <= size)
28.     {
29.         if(ch+1 <= size && dist[heap[ch+1]] < dist[heap[ch]]) ch++;
30.         if(dist[heap[ch]] < dist[t])
31.         {
32.             heap[root] = heap[ch];
33.             map[heap[root]] = root;
34.             root = ch;
35.             ch <<= 1;
36.         }
37.         else break;
38.     }
39.     heap[root] = t;
40.     map[heap[root]] = root;
41. }
43. int heap_pop()
44. {
45.     int t = heap[1];
46.     heap[1] = heap[size--];
47.     map[heap[1]] = 1;
48.     heap_down(1);
49.     return t;
50. }
51. int prime()
52. {
53.     int i, j, k, t, u;
55.     for(i = 1; i <= n; i++)
56.     {
57.         dist[i] = INF;
58.         heap[i] = i;
59.         map[i] = i;
60.         del[i] = 0;
61.     }
62.     size = n;
63.     dist[1] = 0;
65.     for(i = 1; i <= n; i++)
66.     {
67.         u = heap_pop();
68.         del[u] = 1;
69.         for(j = 1; j <= n; j++)
70.         {
71.             if(!del[j] && g[u][j] < dist[j])
72.             {
73.                 dist[j] = g[u][j];
74.                 heap_up(map[j]);
75.             }
76.         }
77.     }
79.     for(i = 1, t = 0; i <= n; i++) t >?= dist[i];
80.     return t;
81. }
82. /*
83. 2
84. 4
85. 0 10 5 200
86. 10 0 200 10
87. 5 200 0 20
88. 200 10 20 0
90. ans=10
91. */
92. int main()
93. {
94.     int i, j, k, t;
96.     scanf("%d", &t);
97.     while(t--)
98.     {
99.         scanf("%d", &n);
100.         for(i = 1; i <= n; i++)
101.         {
102.             for(j = 1; j <= n; j++)
103.             {
104.                 scanf("%d", &g[i][j]);
105.             }
106.             g[i][i] = INF;
107.         }
109.         printf("%d/n", prime());
110.     }
111.     return 0;
112. }