USACO4.4 Shuttle Puzzle【bfs+优化】

直接上$bfs$，每一个状态记录下当前字符串的样子，空格的位置，和走到这个状态的答案。

用空格的位置转移，只有$50pts$

考虑到题目一个性质：$W$只往右走，$B$只往左走，就可以过了。

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
#include<queue>
#include<map>
#include<iostream>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
struct node{
    int pos;
    string s,ans;
};
int n;
queue<node>Q;
map<string,bool>vis;
string aim;
string bfs()
{
    node st;string aim="";
    st.s="";
    for(int i=1;i<=n;i++)
        st.s+='W',aim+='B';
    st.s+='@',aim+='@';
    for(int i=1;i<=n;i++)
        st.s+='B',aim+='W';
    st.pos=n;
    Q.push(st);
    vis[st.s]=1;
    while(!Q.empty())
    {
        node now=Q.front();Q.pop();
        if(now.s==aim)
            return now.ans;
        //cout<<now.s<<endl;
        int idx=now.pos,len=now.s.size();
        node nxt=now;
        if(idx-2>=0&&nxt.s[idx-2]!=nxt.s[idx-1]&&nxt.s[idx-2]=='W')
        {
            nxt.s[idx]=nxt.s[idx-2];
            nxt.s[idx-2]='@';
            //cout<<nxt.s<<endl;//
            char ch='0'+idx-2;
            nxt.ans=now.ans+ch;
            nxt.pos=idx-2;
            if(!vis[nxt.s])
                Q.push(nxt);
            vis[nxt.s]=1;
        }
        nxt=now;
        if(idx-1>=0&&nxt.s[idx-1]=='W')
        {
            nxt.s[idx]=nxt.s[idx-1];
            nxt.s[idx-1]='@';
            //cout<<nxt.s<<endl;//
            char ch='0'+idx-1;
            nxt.ans=now.ans+ch;
            nxt.pos=idx-1;
            if(!vis[nxt.s])
                Q.push(nxt);
            vis[nxt.s]=1;
        }
        nxt=now;
        if(idx+1<len&&nxt.s[idx+1]=='B')
        {
            nxt.s[idx]=nxt.s[idx+1];
            nxt.s[idx+1]='@';
            char ch='0'+idx+1;
            nxt.ans=now.ans+ch;
            nxt.pos=idx+1;
            if(!vis[nxt.s])
                Q.push(nxt);
            vis[nxt.s]=1;
        }
        nxt=now;
        if(idx+2<len&&nxt.s[idx+2]!=nxt.s[idx+1]&&nxt.s[idx+2]=='B')
        {
            nxt.s[idx]=nxt.s[idx+2];
            nxt.s[idx+2]='@';
            char ch='0'+idx+2;
            nxt.ans=now.ans+ch;
            nxt.pos=idx+2;
            if(!vis[nxt.s])
                Q.push(nxt);
            vis[nxt.s]=1;
        }
    }
    return "";
}
int main() 
{
    //freopen("shuttle.in","r",stdin);
    //freopen("shuttle.out","w",stdout);
    scanf("%d",&n);
    string c=bfs();
    //cout<<c<<endl;
    int tot=0;
    for(int i=0;i<c.size();i++)
    {    
        tot++;
        printf("%d",c[i]-'0'+1);
        if(tot==20||i==c.size()-1)
        {
            puts("");
            tot=0;
        }
        else printf(" ");
    }
    return 0;
}