Codeforces Round #139 (Div. 2) D. Snake

Codeforces Round #139 (Div. 2)  D. Snake

A这道题目，A的太艰难了，开始还sb的用dfs + 回溯来搞，是多么的幼稚

bfs+ 判重来求最短路

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
#include <queue>
#include <set>

using namespace std;

const int maxn=20;
//int vis[maxn][maxn][maxn][maxn];
int n,m,css,d[maxn][maxn],len;
set<int> st;

struct Node
{
    int x[9],y[9];
    int step;
};
char mp[maxn][maxn];
queue<Node> q;
int dir[4][2]={{-1,0},{0,-1},{1,0},{0,1}};

bool judge(int x,int y)
{
    if(x>=0&&x<n&&y>=0&&y<m) return 1;
    return 0;
}
int hash(Node a)
{
    int ret=0;
    for(int i=0;i<len;i++) ret=ret*131+a.x[i];
    for(int i=0;i<len;i++) ret=ret*131+a.y[i];
    return ret;
}
int bfs()
{
     for(;!q.empty();q.pop())
     {
         //cout<<"bug"<<endl;
         Node cur=q.front(),tmp;
         int nx,ny;
         css++;
         for(int i=0;i<len-1;i++) d[cur.x[i]][cur.y[i]]=css;
         for(int j,i=0;i<4;i++)
         {
             nx=cur.x[0]+dir[i][0];
             ny=cur.y[0]+dir[i][1];
             if(!judge(nx,ny)||mp[nx][ny]=='#') continue;
             if(d[nx][ny]==css) continue;
             if(mp[nx][ny]=='@') return cur.step+1;

             tmp.x[0]=nx,tmp.y[0]=ny;
             tmp.step=cur.step+1;
             for(int i=0;i<len-1;i++)
               tmp.x[i+1]=cur.x[i],tmp.y[i+1]=cur.y[i];

             int hs=hash(tmp);
             if(st.count(hs)) continue;
             st.insert(hs);

             q.push(tmp);
         }
     }
     return -1;
}
int main()
{
    while(scanf("%d%d",&n,&m)==2)
    {
        for(int i=0;i<n;i++) scanf("%s",mp[i]);
        Node tmp;
        len=0;
        for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
          if(mp[i][j]>='1'&&mp[i][j]<='9')
          {
              char cc=mp[i][j];
              tmp.x[cc-'1']=i,tmp.y[cc-'1']=j;
              if(len < (cc-'1')) len = cc-'1';
          }
        tmp.step=0;
        len++;

        q.push(tmp);
        st.insert(hash(tmp));

        printf("%d\n",bfs());
    }
    return 0;
}