POJ 1176 Party Lamps 稍有玄机

题意：

 

对于一串彩灯，提供四种改变彩灯状态（ON<=>OFF）的操作：a.改变所有彩灯状态；b.改变奇数彩灯状态；c.改变偶数彩灯状态；d.改变3k+1号彩灯状态（1,4,7,10...）。

 

给定彩灯数目，操作次数，和对于某几个彩灯必须为ON、某几个彩灯必须为OFF的要求，问经过给定次数的操作，最终能达到的满足要求的状态有多少种，输出所有满足要求的彩灯状态。

 

原题中操作次数是1<=C<=10000的，如果以此为搜索深度，未免比较可怕。还好这里有点小玄机，可以将搜索次数大大限制。

考虑那四个操作，可以发现有这样的特点：

 

1.操作序列中的各操作是可以交换的，即ab=ba；

2.每两次相同操作效果抵消，即aab=b。

 

据此处理一个操作序列，首先将所有相同的操作归并到一起，得到aa...ab....bc....cd....d的序列；然后将偶数个相同操作消除，奇数个相同操作剩余一个即可，得到一个小于等于4个操作的操作序列，其中每个操作不多于一次。

 

所以可以将操作数限制在4以下，4^4=256，一共有256种最终状态，暴搜即可。以4层搜索为例，每层搜索中求解彩灯状态建立在上层搜索的结果状态上，用一个线性数组存储所有层次结果的话，需要建立起下层结果和上层结果在存储位置上的对应方式，由于每次搜索将结果数扩张四倍，所以得到对应关系为i = (j-1)/4。

 

代码如下：

#include <iostream> #include <string> #include <set> using namespace std; int N,C,num = 1; bool lamps[101],ONs[101],OFFs[101]; bool results[345][101]; set<string> ResSet; void Change_All() { if (num <= 4) { for (int i = 1;i <= N;i++) { results[num][i] = !lamps[i]; } } else { for (int i = 1;i <= N;i++) { results[num][i] = !results[(num-1)/4][i]; } } } void Change_Odds() { if (num <= 4) { for (int i = 1;i <= N;i++) { if (i%2==1) results[num][i] = !lamps[i]; else results[num][i] = lamps[i]; } } else { for (int i = 1;i <= N;i++) { if (i%2==1) results[num][i] = !results[(num-1)/4][i]; else results[num][i] = results[(num-1)/4][i]; } } } void Change_Evens() { if (num <= 4) { for (int i = 1;i <= N;i++) { if (i%2==0) results[num][i] = !lamps[i]; else results[num][i] = lamps[i]; } } else { for (int i = 1;i <= N;i++) { if (i%2==0) results[num][i] = !results[(num-1)/4][i]; else results[num][i] = results[(num-1)/4][i]; } } } void Change_3K1() { if (num <= 4) { for (int i = 1;i <= N;i++) { if (i%3==1) results[num][i] = !lamps[i]; else results[num][i] = lamps[i]; } } else { for (int i = 1;i <= N;i++) { if (i%3==1) results[num][i] = !results[(num-1)/4][i]; else results[num][i] = results[(num-1)/4][i]; } } } bool Check() { for (int i = 1;i <= N;i++) { if (ONs[i]==true&&results[num][i]==false) { return false; } if (OFFs[i]==true&&results[num][i]==true) { return false; } } return true; } int main() { //int C; memset(lamps,true,101); memset(ONs,false,101); memset(OFFs,false,101); memset(results,false,sizeof(results)); for (int i = 1;i <= 4;i++) { memset(results[i],true,101); } void (*pfcn[4])(); pfcn[0] = Change_All; pfcn[1] = Change_Odds; pfcn[2] = Change_Evens; pfcn[3] = Change_3K1; cin>>N>>C; if(C>4) { C%=2; if (C==1) { C=3; } else { C=4; } } int temp; while (1) { cin>>temp; if (temp == -1) { break; } ONs[temp] = true; } while (1) { cin>>temp; if (temp == -1) { break; } OFFs[temp] = true; } int loop = 1; while (C--) { loop *= 4; for (int i = 0;i < loop;i++) { pfcn[i%4](); if (C==0) { if (Check()) { string str; for (int j = 1;j <= N;j++) { str.push_back(results[num][j]?'1':'0'); //cout<<results[num][j]?1:0; } ResSet.insert(str); } } ++num; } } for (set<string>::iterator it = ResSet.begin();it != ResSet.end();it++) { cout<<*it<<endl; } }