uvaoj 321 - The New Villa

传送:点击打开链接

一道容易题，弄了我好久啊，各种粗心，郁闷死了！

题意：给三个整数，分别表示：房间数，图的连接情况，每个房间能控制的灯。输出从起点到终点的方式。

搜索很好理解，关键是判重。

可以将灯的情况抽象成01串，人处在的房间号放在房间数+1的位置

例如第一个样例 3个房间初始就可以用 1 0 0 1表示，终点就是 0 0 1 3

再例如10 起点就是 1 0 0 0 0 0 0 0 0 0 1 终点就是0 0 0 0 0 0 0 0 0 1 10

然后再通过转换成10进制对应判重就可以了。

注意考虑房间只有1个的时候。

#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
#define MAXN 50000      //其实不用这么大,2^10-1+10*2^10应该就够了
int map[12][12];
int light[12][12];
char vis[MAXN];
int n,d,k;
int end;
struct node
{
    int dis;
    int light[12];
}q[MAXN];
int path[MAXN];
int fa[MAXN];
int ok;
int base[]={0,1,1<<1,1<<2,1<<3,1<<4,1<<5,1<<6,1<<7,1<<8,1<<9,1<<10};
    //通过二进制转换成十进制将每个房间的灯+人的位置的情况与某个整数一一对应
int hash(int *a)
{
    int s=0;
    for(int i=1;i<=n+1;i++)
    {
        s+=(a[i]*base[i]);
    }
    if(s==end) ok=1;
    return s;
}
int bfs()
{
    int rear=1,front=1;
    memset(q[1].light,0,sizeof(q[1].light));
    vis[1+base[n+1]]=1;
    q[1].light[1]=1;
    q[1].light[n+1]=1;
    q[1].dis=0;
    fa[1]=0;
    node tmp;
    rear++;
    while(front<rear)
    {
        tmp=q[front];
        for(int i=1;i<=n;i++)   //把当前房间能控制的灯打开或关上
        {
            if(light[tmp.light[n+1]][i]&&tmp.light[n+1]!=i)
            {
                node tt=tmp;
                tt.light[i]=1-tt.light[i];
                int x=hash(tt.light);
                if(!vis[x])
                {
                    vis[x]=1;
                    tt.dis++;
                    fa[rear]=front;
                    if(!tt.light[i]) path[rear]=-i*1000;     //开关灯保存在百位和千位，负数表示关(纯属个人喜好=_=)
                    else path[rear]=i*1000;
                    q[rear++]=tt;

                    if(ok) {return rear-1;}
                }
            }
        }

        for(int i=1;i<=n;i++)   //能走到的所有房间
        {
            if(map[tmp.light[n+1]][i]&&tmp.light[i])
            {
                node tt=tmp;
                tt.light[n+1]=i;
                int x=hash(tt.light);
                if(!vis[x])
                {
                    vis[x]=1;
                    tt.dis++;
                    fa[rear]=front;
                    path[rear]=i;       //走到某个房间保存在个位和十位
                    q[rear++]=tt;

                    if(ok) {return rear-1;}
                }
            }
        }

        front++;
    }
    return -1;
}
void print(int k)
{
    if(!fa[k]) return;
    print(fa[k]);
    if(path[k]<1000&&path[k]>0)
        printf("- Move to room %d.\n",path[k]);
    else if(path[k]>0)
        printf("- Switch on light in room %d.\n",path[k]/1000);
    else
        printf("- Switch off light in room %d.\n",(-path[k])/1000);

}
int main()
{
    int cas=1;
    int a,b;
    while(scanf("%d%d%d",&n,&d,&k))
	{
	    ok=0;

		if(n==0&&d==0&&k==0) break;
		memset(map,0,sizeof(map));
		memset(light,0,sizeof(light));
		memset(vis,0,sizeof(vis));
        for(int i=1;i<=d;i++)
        {
            scanf("%d%d",&a,&b);
            map[a][b]=map[b][a]=1;
        }
        for(int i=1;i<=k;i++)
        {
            scanf("%d%d",&a,&b);
            light[a][b]=1;
        }
        printf("Villa #%d\n",cas++);
        if(n==1) {printf("The problem can be solved in 0 steps:\n\n");continue;}
        //注意考虑房间只有1个的情况！
        end=base[n]+n*base[n+1];        
        int flag=bfs();

        if(flag<0) printf("The problem cannot be solved.\n");
        else
        {
            printf("The problem can be solved in %d steps:\n",q[flag].dis);
            print(flag);
        }
        putchar(10);
	}
    return 0;
}