压入重标记（预流推进）

1. #include <stdio.h>
2. #include <string.h>
4. const int max = 54 ;
5. const long MAX = 200000000 ;
7. int N, M, R ;
8. long map[max][max] ;    //权值
9. int src, dis ;                      //源点和汇点
11. long star[max][3] ;
13. long ef[max] ;      //余流
14. int height[max] ;   //高度
16. long sum ;          //最大流，即最小截值
18. void Init(void)
19. {
20. //  freopen("starwar.txt", "r", stdin);
22.     int loop ;
23.     int x , y ;
25.     memset(map, 0, sizeof(map)) ;
27.     scanf("%d %d %d", &N, &M, &R) ;
29.     dis = N + 1 ;
31.     for ( loop = 1; loop <= N; loop++ )
32.     {
33.         scanf("%d %d %d", 
34.             &star[loop][0], &star[loop][1], &star[loop][2]) ;
35.     }
36.     
37.     for ( loop = 1; loop <= M; loop++ )
38.     {
39.         scanf("%d %d", &x, &y);
40.         map[x][y] = map[y][x] =
41.             ( (star[x][0] - star[y][0]) * (star[x][0] - star[y][0])
42.             + (star[x][1] - star[y][1]) * (star[x][1] - star[y][1])
43.             + (star[x][2] - star[y][2]) * (star[x][2] - star[y][2]) ) ;
44.     }
46.     for ( loop = 1; loop <= N; loop++ )
47.     {
48.         if ( (star[loop][0] * star[loop][0] + star[loop][1] * star[loop][1]
49.             + star[loop][2] * star[loop][2]) > (R * R) )
50.         {
51.             map[loop][dis] = MAX ;
52.         }
53.     }
55.     src = 0 ;
56.     for ( loop = 0; loop <= N + 1; loop++ )
57.     {
58.         height[loop] = 0 ;
59.     }
60.     ef[src] = MAX ;
61.     ef[dis] = -MAX ;
63.     height[src] = N + 1 ;
65.     sum = 0 ;
66. }
68. void PushRelabel(void)
69. {
70.     long queue[max] , head , tail ;
71.     queue[0] = src ;
72.     head = 0 ;
73.     tail = 1 ;
75.     long u , v , p ;
76.     int i ;
78.     while( head != tail )
79.     {
80.         u = queue[head++] ;
82.         if ( head == max )
83.         {
84.             head = 0 ;
85.         }
87.         for ( i = 0 ; i <= N + 1 ; i++ )
88.         {
89.             v = i ;
90.             
91.             /*
92.             *   压入操作
93.             */
94.             p = (map[u][v] < ef[u] ? map[u][v] : ef[u]) ;
96.             if ( (p > 0) && (u == src || height[u] == height[v] + 1) )
97.             {
98.                 map[u][v] -= p ;
99.                 map[v][u] += p ;
101.                 if ( v == dis )
102.                 {
103.                     sum += p ;
104.                 }
106.                 ef[u] -= p ;
107.                 ef[v] += p ;
109.                 if ( v != src && v != dis )
110.                 {
111.                     queue[tail++] = v ;
113.                     if ( tail == max )
114.                     {
115.                         tail = 0 ;
116.                     }
117.                 }
118.             }
119.         }
121.         /*
122.         *   重标记
123.         */
124.         if ( u != src && u != dis && ef[u] > 0 )
125.         {
126.             height[u]++;
127.             queue[tail++] = u ;
129.             if ( tail == max )
130.             {
131.                 tail = 0 ;
132.             }
133.         }
134.     }
135. }
137. int main(void)
138. {
139.     Init();
140.     PushRelabel();
142.     printf("%ld/n", sum) ;
144.     return 0;
145. }