CSU 1259 bfs找最短路

题目大意：

不想介绍，题目链接：http://acm.csu.edu.cn/OnlineJudge/problem.php?id=1259

bfs求最短路。

这里因为2-9，到达同样的点不计步数，那我们不能每次bfs都遍历一个图找到所有相同的点，所以这里用vector[10]，保存对应的2-9的相同的节点

题目本身比较简单，只是考虑的东西有点多，慢慢写就行了

  1 #include <cstdio>

  2 #include <cstring>

  3 #include <iostream>

  4 #include <vector>

  5 #include <queue>

  6 using namespace std;

  7 #define N 105

  8 int dir[4][2] = {{1,0},{-1,0},{0,1},{0,-1}};

  9 int dp[N][N],n,visit[N][N];

 10 int mat[N][N];

 11 struct Node{

 12     int x,y;

 13     Node(int x=0,int y=0):x(x),y(y){}

 14     bool operator==(const Node &m)const{

 15         return x == m.x && y == m.y;

 16     }

 17 };

 18 Node st , la;

 19 vector<Node> G[10];

 20 queue<Node> q;

 21 

 22 bool ok(int x,int y){

 23     if(x>=1 && x<=n && y>=1 && y<=n && mat[x][y] != 1)

 24         return true;

 25     return false;

 26 }

 27 

 28 void bfs()

 29 {

 30     q.push(st);

 31     visit[st.x][st.y] = 1;

 32     dp[st.x][st.y] = 0;

 33     while(!q.empty()){

 34         Node u = q.front();

 35        //cout<<"test: "<<u.x<<"  "<<u.y<<"  "<<dp[x][y]<<endl;

 36         q.pop();

 37         if(u == la) continue;

 38         if(mat[u.x][u.y] == 0){

 39             for(int i=0;i<4;i++){

 40                 int xx = u.x+dir[i][0];

 41                 int yy = u.y+dir[i][1];

 42                 if(ok(xx,yy)){

 43                     if(dp[xx][yy] > dp[u.x][u.y] + 1){

 44                         dp[xx][yy] = dp[u.x][u.y] + 1;

 45                         if(!visit[xx][yy]){

 46                             visit[xx][yy]=1;

 47                             q.push(Node(xx,yy));

 48                         }

 49                     }

 50                 }

 51             }

 52         }

 53         else{

 54             int t = mat[u.x][u.y];

 55             for(int i=0;i<(int)G[t].size();i++){

 56                 Node m = G[t][i];

 57                 if(m == u) continue;

 58                 if(dp[m.x][m.y] > dp[u.x][u.y])

 59                 {

 60                     dp[m.x][m.y] = dp[u.x][u.y];

 61                     if(!visit[m.x][m.y]){

 62                         visit[m.x][m.y] = 1;

 63                         q.push(m);

 64                     }

 65                 }

 66             }

 67             for(int i=0;i<4;i++){

 68                 int xx = u.x+dir[i][0];

 69                 int yy = u.y+dir[i][1];

 70                 if(ok(xx,yy)){

 71                     if(dp[xx][yy] > dp[u.x][u.y] + 1){

 72                         dp[xx][yy] = dp[u.x][u.y] + 1;

 73                         if(!visit[xx][yy]){

 74                             visit[xx][yy]=1;

 75                             q.push(Node(xx,yy));

 76                         }

 77                     }

 78                 }

 79             }

 80         }

 81     }

 82 }

 83 

 84 int main()

 85 {

 86     freopen("test.in","rb",stdin);

 87     char a;

 88     while(~scanf("%d",&n)){

 89         //init

 90         for(int i=1;i<10;i++)G[i].clear();

 91         memset(dp,0x3f,sizeof(dp));

 92         memset(visit,0,sizeof(visit));

 93 

 94         for(int i=1;i<=n;i++){

 95             for(int j=1;j<=n;j++){

 96                 cin>>a;

 97                 if(a == 'S')

 98                 {

 99                     mat[i][j] = 0;

100                     st = Node(i,j);

101                 }

102                 else if(a == 'E')

103                 {

104                     mat[i][j] = 0;

105                     la = Node(i,j);

106                 }

107                 else if(a == '0' || a == '1'){

108                     mat[i][j] = 0 + a - '0';

109                 }

110                 else{

111                     mat[i][j] = 0 + a - '0';

112                     G[mat[i][j]].push_back(Node(i,j));

113                 }

114             }

115         }

116 

117         bfs();

118         if(dp[la.x][la.y] == 0x3f3f3f3f){

119             puts("Oh No!");

120         }

121         else{

122             printf("%d\n",dp[la.x][la.y]);

123         }

124     }

125     return 0;

126 }