pku1103

1. #include <stdio.h>
2. #include <memory.h>
4. const int MAXN = 50 ;
5. const int MAX_DIST = 1000 ;
7. int iDist[MAXN] ;
8. bool bMap[MAXN][MAXN] , bVisit[MAXN] ;
9. int iN , guardPos ;
11. void dfs( int p )
12. {
13.     int i ;
14.     bVisit[p] = true ;
16.     for ( i = 0 ; i < iN ; i++ )
17.         if ( i != guardPos && bMap[p][i] && !bVisit[i] )
18.         {
19.             dfs(i) ;
20.         }
21. }
23. void spfa( int iSrc )
24. {
25.     int iQueue[MAXN], head , tail , i , u , v ;
27.     for ( i = 0 ; i < iN ; i++ )
28.     {
29.         iDist[i] = MAX_DIST ;
30.     }
31.     
32.     head = 0 ;
33.     tail = 1 ;
34.     iQueue[0] = iSrc ;
35.     iDist[iSrc] = 0 ;
37.     while ( head != tail )
38.     {
39.         u = iQueue[head] ;
40.         head = ( head + 1 ) % MAXN ;
42.         for ( v = 0 ; v < iN ; v++ )
43.         {
44.             if ( bMap[v][u] && iDist[u] + 1 < iDist[v] )
45.             {
46.                 iDist[v] = iDist[u] + 1 ;
47.                 iQueue[tail] = v ;
48.                 tail = ( tail + 1 ) % MAXN ;
49.             }
50.         }
51.     }
52. }
54. int main()
55. {
56.     int etpos , ix , iy ;
58.     scanf("%d %d", &iN, &etpos) ;
60.     memset( bMap, 0, sizeof(bMap) ) ;
62.     while ( scanf("%d %d", &ix, &iy) != EOF )
63.     {
64.         bMap[ix][iy] = true ;
65.     }
66.     
67.     spfa( etpos ) ;    //计算出从ET所在点到其他点的最短路径
69.     int i , min ;
70.     bool bX[MAXN] = { false } ;
72.     while ( true )
73.     {
74.         min = MAX_DIST ;
76.         for ( i = 0 ; i < iN ; i++ )
77.             if ( i != etpos && bX[i] == false )
78.             {
79.                 if ( min > iDist[i] )
80.                 {
81.                     guardPos = i ;
82.                     min = iDist[i] ;
83.                 }
84.             }
86.         if ( min == MAX_DIST )
87.         {
88.             printf("Put guards in room 0./n") ;
89.             break ;         
90.         }
92.         memset( bVisit, 0, sizeof(bVisit) ) ;
94.         bX[guardPos] = true ;
96.         dfs(0) ;
98.         if ( !bVisit[etpos] )  //若无法访问到ET所在的点，则guardPos为所求监控点
99.         {
100.             printf("Put guards in room %d./n", guardPos) ;
101.             break ;
102.         }
103.     }
105.     return 0 ;
106. }