UVa-1600 Patrol Robot
1 #include <bits/stdc++.h> 2 3 using namespace std; 4 5 struct Node 6 { 7 int r,c,g; 8 Node(int r,int c,int g): r(r),c(c),g(g) {} 9 Node() {} 10 }; 11 12 int m[22][22]; 13 14 int rt,ct; 15 int k; 16 void read() 17 { 18 memset(m,0,sizeof(m)); 19 cin >> rt >> ct >> k; 20 for(int i = 0;i < rt;i ++) 21 for(int j = 0; j < ct;j ++) 22 cin >> m[i][j]; 23 } 24 25 const int dr[] = {1,0,0,-1}; 26 const int dc[] = {0,1,-1,0}; 27 28 int d[22][22][22]; 29 bool limit(Node v) 30 { 31 return v.r>=0&&v.r<rt&&v.c>=0&&v.c<ct&&v.g<=k; 32 } 33 int solve() 34 { 35 memset(d,-1,sizeof(d)); 36 queue<Node> q; 37 q.push(Node(0,0,m[0][0])); 38 d[0][0][m[0][0]] = 0; 39 while(!q.empty()) 40 { 41 Node u = q.front();q.pop(); 42 if(u.r==rt-1&&u.c==ct-1) return d[u.r][u.c][u.g]; 43 for(int i = 0; i < 4;i ++) 44 { 45 Node v = u;v.r += dr[i];v.c += dc[i]; 46 if(limit(v)&&m[v.r][v.c]==0) v.g = 0; 47 if(limit(v)&&m[v.r][v.c]==1) v.g ++; 48 if(limit(v)&&d[v.r][v.c][v.g]<0) 49 { 50 d[v.r][v.c][v.g] = d[u.r][u.c][u.g] + 1; 51 q.push(v); 52 } 53 } 54 } 55 return -1; 56 } 57 58 void print(int rnt) 59 { 60 cout << rnt << endl; 61 } 62 int main() 63 { 64 int T; 65 cin >> T; 66 while(T --) 67 { 68 read(); 69 int rnt = solve(); 70 print(rnt); 71 } 72 return 0; 73 }