pku 1161 Walls

1. #include <cstdio>
2. #include <cstring>
4. const int MAXN = 252 ;
5. const int MAX_INF = 50000 ;
7. int gMember[MAXN] ;
8. int region[MAXN][MAXN] ;
9. bool place[MAXN][MAXN] ;
11. void Floyd ( int n )
12. {
13.     int i , j , k ;
14.         
15.     for ( k = 1 ; k <= n ; k++ )
16.     {
17.         for ( i = 1 ; i <= n ; i++ )
18.             for ( j = 1 ; j <= n ; j++ )
19.             {
20.                 if ( region[i][j] > region[i][k] + region[k][j] )
21.                 {
22.                     region[i][j] = region[i][k] + region[k][j] ;
23.                 }
24.             }
25.     }
26. }
28. int SearchPos( int m , int l )
29. {
30.     int i , j , k , min , sum , temp ;
32.     min = MAX_INF ;
33.     for ( k = 1 ; k <= m ; k++ )
34.     {
35.         sum = 0 ;
36.         for ( i = 0 ; i < l ; i++ )
37.         {
38.             temp = MAX_INF ;
39.             for ( j = 1 ; j <= m ; j++ )
40.             {
41.                 if ( place[gMember[i]][j] && temp > region[j][k] )
42.                 {
43.                     temp = region[j][k] ;
44.                 }
45.             }
46.             sum += temp ;
47.         }
49.         if ( min > sum )
50.         {
51.             min = sum ;
52.         }
54.     }
56.     return min ;
57. }
59. int main()
60. {
61.     int m , n , l , i , j , k , num , t ;
62.     int pos[MAXN] , temp[MAXN][MAXN] ;
64.     scanf("%d", &m) ;
66.     scanf("%d", &n) ;
68.     scanf("%d", &l) ;
69.     for ( i = 0 ; i < l ; i++ )
70.     {
71.         scanf("%d", &gMember[i]) ;
72.     }
74.     memset(temp, 0, sizeof(temp)) ;
75.     memset(place, 0, sizeof(place)) ;
76.     
77.     for ( i = 1 ; i <= m ; i++ )
78.         for ( j = 1 ; j <= m ; j++ )
79.         {
80.             if ( i == j )
81.                 region[i][j] = 0 ;
82.             else 
83.                 region[i][j] = MAX_INF ;
84.         }
86.     for ( i = 1 ; i <= m ; i++ )
87.     {
88.         scanf("%d", &num) ;
89.         for ( j = 0 ; j < num ; j++ )
90.         {
91.             scanf("%d", &pos[j]) ;
92.             place[pos[j]][i] = true ;
93.         }
94.         
95.         for ( k = 0 ; k < num ; k++ )
96.         {
97.             j = pos[k] ;
98.             if ( k == num - 1 )
99.             {
100.                 if ( temp[j][pos[0]] > 0 )
101.                 {
102.                     t = temp[j][pos[0]] ;
103.                     region[i][t] = region[t][i] = 1 ;
104.                 }
105.                 else 
106.                 {
107.                     temp[j][pos[0]] = temp[pos[0]][j] = i ;
108.                 }
109.             }
110.             else
111.             {
112.                 if ( temp[j][pos[k + 1]] > 0 )
113.                 {
114.                     t = temp[j][pos[k + 1]] ;
115.                     region[i][t] = region[t][i] = 1 ;
116.                 }
117.                 else 
118.                 {
119.                     temp[j][pos[k + 1]] = temp[pos[k + 1]][j] = i ;
120.                 }
121.             }
122.         }
123.     }
125.     Floyd( m ) ;
127.     int ans = SearchPos( m , l ) ;
129.     printf("%d/n", ans) ;
131.     return 0 ;
132. }