一道特别经典的枚举题目,核心思想并非仅仅是单纯的枚举,而是通过枚举局部的可能来确定整体的情况,因为在单纯的枚举中,含有大量的不符合要求的情况。当第一行的开关按法唯一确定时,为了迎合题目要求,之后所有按法都唯一确定,当最后一回合按下后,倒数第二行灯已全灭,只需判断最后一行是否全灭即可。通过枚举出局部的情况而确定全局的情况,这种思想大大减少了枚举的盲目性。
Talk is cheap. Show me the code:
#include <iostream>
#include <memory>
#include <cstring>
using namespace std;
void SetBit(char &c,int k)
{
c|=(1<<k);
}
int GetBit(char c,int k)
{
return 1&(c>>k);
}
void Filp(char &c,int k)
{
c^=(1<<k); //与1取异或反转
}
void Light_Read(char oriLight[])
{
int s;
for(int i=0;i<5;i++)
{
for(int j=0;j<6;j++)
{
cin>>s;
if(s)
SetBit(oriLight[i],j);
}
}
return;
}
void Output(char result[],int n)
{
cout<<"PUZZLE #"<<n<<endl;
for(int i=0;i<5;i++)
{
for(int j=0;j<6;j++)
{
if(!j)
cout<<GetBit(result[i],j);
else
cout<<" "<<GetBit(result[i],j);
}
cout<<endl;
}
}
void Lights(int count)
{
char oriLight[5];
char Light[5];
char result[5];
char switchs;
memset(oriLight,0,sizeof(oriLight));
Light_Read(oriLight);
for(int i=0;i<64;i++) //枚举第一行所有情况
{
switchs=i;
memcpy(Light,oriLight,sizeof(oriLight));
for(int j=0;j<5;j++)
{
result[j]=switchs;
for(int k=0;k<6;k++)
{
if(GetBit(switchs,k))
{
if(k>0)
{
Filp(Light[j],k-1);
}
Filp(Light[j],k);
if(k<5)
{
Filp(Light[j],k+1);
}
}
}
if(j<4)
{
Light[j+1]^=switchs; //十分巧妙的异或运算
}
switchs=Light[j]; //为保证第j行全灭,下一行方案必定为此
if(j==4&&!Light[j])
{
Output(result,count);
break;
}
}
}
}
int main(void)
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
Lights(i);
}
return 0;
}
代码参考了北京大学郭炜老师的在慕课上的算法课程写出来的,也不禁感叹位运算和几种逻辑运算的强大和自己的无知。路漫漫其修远兮,吾将上下而求索。