【BFS/DFS/YY】派对灯

[luogu1468]派对灯

题目描述

在IOI98的节日宴会上，我们有N(10<=N<=100)盏彩色灯，他们分别从1到N被标上号码。这些灯都连接到四个按钮：

按钮1：当按下此按钮，将改变所有的灯：本来亮着的灯就熄灭，本来是关着的灯被点亮。

按钮2：当按下此按钮，将改变所有奇数号的灯。

按钮3：当按下此按钮，将改变所有偶数号的灯。

按钮4：当按下此按钮，将改变所有序号是3*K+1(K>=0)的灯。例如：1,4,7...

一个计数器C记录按钮被按下的次数。当宴会开始，所有的灯都亮着，此时计数器C为0。

你将得到计数器C(0<=C<=10000)上的数值和经过若干操作后某些灯的状态。写一个程序去找出所有灯最后可能的与所给出信息相符的状态，并且没有重复。

输入输出格式

输入格式：

不会有灯会在输入中出现两次。

第一行: N。

第二行: C最后显示的数值。

第三行: 最后亮着的灯,用一个空格分开，以-1为结束。

第四行: 最后关着的灯,用一个空格分开，以-1为结束。

输出格式：

每一行是所有灯可能的最后状态(没有重复)。每一行有N个字符，第1个字符表示1号灯，最后一个字符表示N号灯。0表示关闭，1表示亮着。这些行必须从小到大排列（看作是二进制数）。

如果没有可能的状态，则输出一行'IMPOSSIBLE'。

输入输出样例

输入样例#1：

输出样例#1：

说明

在这个样例中，有三种可能的状态：

所有灯都关着

1,4,7,10号灯关着，2,3,5,6,8,9亮着。

1,3,5,7,9号灯关着，2, 4, 6, 8, 10亮着。

翻译来自NOCOW

USACO 2.2

试题分析：第一眼看：很暴力！但再看一眼，发现我们只需要枚举第一次的操作，然后看是否满足条件，条件分以下几种：

①变换完不满足要求 X

②变换完满足要求但最后调完会变成不满足要求的：(C-k)%2!=0

③调的次数大于C的

以二进制形式枚举1到16，每位表示这个操作用不用

然后经过处理不合法，在用结构体排一下序(因为方便)就AC了

代码

#include<iostream>

#include<cstring>

#include<cstdio>

#include<queue>

#include<stack>

#include<vector>

#include<algorithm>

//#include<cmath>

using namespace std;

const int INF = 9999999;

#define LL long long

inline int read(){

	int x=0,f=1;char c=getchar();

	for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;

	for(;isdigit(c);c=getchar()) x=x*10+c-'0';

	return x*f;

}

int N,C;

bool light[1001];

bool dark[1001];

bool cnt[1001];

int ta;

struct data{

    int seq[105];

}ans[5001];

bool cmp(data a,data b){

	for(int i=1;i<=N;i++){

		if(a.seq[i]!=b.seq[i]) return a.seq[i]<b.seq[i];

	}

}

int main(){

	//freopen(".in","r",stdin);

	//freopen(".out","w",stdout);

	N=read(),C=read();

	int tmp;

	tmp=read();

	while(tmp!=-1){

		light[tmp]=1;

	    tmp=read();

	}

	tmp=read();

	while(tmp!=-1){

		dark[tmp]=1;

		tmp=read();

	}

	for(int i=1;i<=16;i++){

		memset(cnt,1,sizeof(cnt));

		int k=0;

		if(i&(1<<0)){k++;for(int j=1;j<=N;j++) cnt[j]=1-cnt[j];}

		if(i&(1<<1)){k++;for(int j=1;j<=N;j+=2) cnt[j]=1-cnt[j];}

		if(i&(1<<2)){k++;for(int j=2;j<=N;j+=2) cnt[j]=1-cnt[j];}

		if(i&(1<<3)) {k++;for(int j=0;3*j+1<=N;j++) cnt[3*j+1]=1-cnt[3*j+1];}

		bool flag=true;

		for(int i=1;i<=N;i++){

			if(cnt[i]!=1&&light[i]==1){

				flag=false;

				break;

			}

		}

		for(int i=1;i<=N;i++){

			if(cnt[i]!=0&&dark[i]==1){

				flag=false;

				break;

			}

		}

		if(!flag) continue;

		if((C-k)%2!=0||k>C) continue;

		++ta;

		for(int j=1;j<=N;j++)

		    ans[ta].seq[j]=cnt[j];

	}

	if(!ta){

		puts("IMPOSSIBLE");

		return 0;

	}

	sort(ans+1,ans+ta+1,cmp);

	for(int i=1;i<=ta;i++){

		for(int j=1;j<=N;j++)

		    printf("%d",ans[i].seq[j]);

		printf("\n");

	}

	return 0;

}