CSU 1259 bfs找最短路

题目大意：

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

bfs求最短路。

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

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

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
#define N 105
int dir[4][2] = {{1,0},{-1,0},{0,1},{0,-1}};
int dp[N][N],n,visit[N][N];
int mat[N][N];
struct Node{
    int x,y;
    Node(int x=0,int y=0):x(x),y(y){}
    bool operator==(const Node &m)const{
        return x == m.x && y == m.y;
    }
};
Node st , la;
vector<Node> G[10];
queue<Node> q;

bool ok(int x,int y){
    if(x>=1 && x<=n && y>=1 && y<=n && mat[x][y] != 1)
        return true;
    return false;
}

void bfs()
{
    q.push(st);
    visit[st.x][st.y] = 1;
    dp[st.x][st.y] = 0;
    while(!q.empty()){
        Node u = q.front();
       //cout<<"test: "<<u.x<<"  "<<u.y<<"  "<<dp[x][y]<<endl;
        q.pop();
        if(u == la) continue;
        if(mat[u.x][u.y] == 0){
            for(int i=0;i<4;i++){
                int xx = u.x+dir[i][0];
                int yy = u.y+dir[i][1];
                if(ok(xx,yy)){
                    if(dp[xx][yy] > dp[u.x][u.y] + 1){
                        dp[xx][yy] = dp[u.x][u.y] + 1;
                        if(!visit[xx][yy]){
                            visit[xx][yy]=1;
                            q.push(Node(xx,yy));
                        }
                    }
                }
            }
        }
        else{
            int t = mat[u.x][u.y];
            for(int i=0;i<(int)G[t].size();i++){
                Node m = G[t][i];
                if(m == u) continue;
                if(dp[m.x][m.y] > dp[u.x][u.y])
                {
                    dp[m.x][m.y] = dp[u.x][u.y];
                    if(!visit[m.x][m.y]){
                        visit[m.x][m.y] = 1;
                        q.push(m);
                    }
                }
            }
            for(int i=0;i<4;i++){
                int xx = u.x+dir[i][0];
                int yy = u.y+dir[i][1];
                if(ok(xx,yy)){
                    if(dp[xx][yy] > dp[u.x][u.y] + 1){
                        dp[xx][yy] = dp[u.x][u.y] + 1;
                        if(!visit[xx][yy]){
                            visit[xx][yy]=1;
                            q.push(Node(xx,yy));
                        }
                    }
                }
            }
        }
    }
}

int main()
{
    freopen("test.in","rb",stdin);
    char a;
    while(~scanf("%d",&n)){
        //init
        for(int i=1;i<10;i++)G[i].clear();
        memset(dp,0x3f,sizeof(dp));
        memset(visit,0,sizeof(visit));

        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                cin>>a;
                if(a == 'S')
                {
                    mat[i][j] = 0;
                    st = Node(i,j);
                }
                else if(a == 'E')
                {
                    mat[i][j] = 0;
                    la = Node(i,j);
                }
                else if(a == '0' || a == '1'){
                    mat[i][j] = 0 + a - '0';
                }
                else{
                    mat[i][j] = 0 + a - '0';
                    G[mat[i][j]].push_back(Node(i,j));
                }
            }
        }

        bfs();
        if(dp[la.x][la.y] == 0x3f3f3f3f){
            puts("Oh No!");
        }
        else{
            printf("%d\n",dp[la.x][la.y]);
        }
    }
    return 0;
}