845. 八数码

Posted 2021-06-02 幽殇默

https://www.acwing.com/problem/content/847/

#include<cstdio>
#include<iostream>
#include<string>
#include<unordered_map>
#include<queue>
using namespace std;
int dx[4]={-1,0,1,0};
int dy[4]={0,1,0,-1};
int bfs(string state)
{
    queue<string> q;  q.push(state);
    unordered_map<string,int> d;  d[state]=0;
    string end="12345678x";
    while(!q.empty())
    {
        string t=q.front(); q.pop();
        if(t==end)
        {
            return d[t];
        }
        int distance=d[t];
        int k=t.find('x');//找到x的位置
        int x=k/3,y=k%3;
        for(int i=0;i<4;i++)
        {
            int a=x+dx[i];
            int b=y+dy[i];
            if(a>=0&&a<3&&b>=0&&b<3)
            {
                swap(t[a*3+b],t[k]);
                if(!d.count(t))  d[t]=distance+1,q.push(t); 
                //找t的个数，如果为0说明没有进去过
                swap(t[a*3+b],t[k]);
            }
        }
    }
    return -1;
}
int main(void)
{
    string a,state;
    for(int i=0;i<9;i++) cin>>a,state+=a;
    cout<<bfs(state)<<endl;
    return 0;
}