【JZOJ B组】【GDKOI2003】分球

Description

　　在一个装满财宝的屋子里，有2N个盒子排成一排。除了两个相邻的空盒外，其余的每个盒子里都装有一个金球或者一个银球，总共有N-1个金球和N-1个银球。以下是一个N=5时的例子，G表示金球，S表示银球：
　　
　　任意两个相邻的非空的盒子里的球可以移动到两个相邻的空盒中，移动不能改变这两个球的排列顺序。写一个程序，用最少的移动次数把所有的金球都移到所有银球的左边。

Input

　　输入文件包含多组数据。第一行为K，表示数据的总数。
　　每组数据的第一行是N(3<=N<=7)，第二行是2N个盒子的初始状态。金球用a表示，银球用b表示，空盒用空格表示。每两组相邻数据用空行隔开。

Output

　　对于每一组数据，若无解则输出一行-1，若有解，输出最少移动次数，相邻的两组数据用一个空行隔开。

Sample Input

3
3
abab

5
abba abab

6
a babbababa

Sample Output

-1

3

4

Data Constraint

k<5

思路

考试的时候没做出来
这题其实一个bfs就行了。
首先我们按三进制状压。然后枚举转移，判断是否到达最终状态即可。

代码

#include 
#include 
#include 
#include 
using namespace std;
int k,n,head,tail,x,p,f[6563*9*16],a[16],b[16],w[6563*9][16];
char ch;
bool check(int *x)
{
    int y=0;
    for (int i=1;i<=n+1;i++)
    {
        y+=x[i]&1;
        if (y==n-1) return 1;
        if (x[i]==2) return 0;
    }
}
void get_num(int *x)
{
    int o=0; p=0;
    for (int i=1;i<=2*n;i++)
        if (x[i]==0&&o==0) o=1;
        else p=p*3+x[i];
}
int bfs()
{
    if (check(w[1])) return 0;
    head=0; tail=1;
    get_num(w[1]);
    f[p]=k;
    while (headmemcpy(a,w[head],sizeof(a)); memcpy(b,a,sizeof(b)); b[0]++;
        for(int i=1; i<=2*n-1; i++) if(a[i]==0) 
        { 
            x=i; break; 
        }
        for(int i=1; i<=2*n-1; i++) 
            if (a[i]!=0&&a[i+1]!=0)
            {
                b[x]=a[i]; b[x+1]=a[i+1]; b[i]=b[i+1]=0; get_num(b);
                if (f[p]!=k)
                {
                    if (check(b)) return b[0];
                    tail++;
                    f[p]=k; 
                    memcpy(w[tail],b,sizeof(w[tail]));
                }
                b[i]=a[i]; b[i+1]=a[i+1];
            }
    }
    return -1;
}
int main()
{
    scanf("%d",&k);
    for(;k;k--)
    {
        scanf("%d",&n);
        getchar();
        for(int i=1; i<=2*n; i++)
        {
            ch=getchar();
            w[1][i]=(ch==' '?0:ch-96);
        }
        printf("%d\n",bfs());
    }
}