[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'。
输入输出样例
10 1 -1 7 -1
0000000000 0101010101 0110110110
说明
在这个样例中，有三种可能的状态：
所有灯都关着
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("
");
}
return 0;
}