枚举算法

1.概念

  • 基于逐个尝试答案的一种问题求解策略
    例:求小于N的最大素数
    –找不到公式,只能一个个试一试。

2.例题
POJ1222
题意:有一个5 * 6的矩阵,每个位置表示灯,1表示灯亮,0表示灯灭。
然后如果选定位置i,j点击,则位置i,j和其上下左右的灯的状态都会反转。
现在要你求出一个5 * 6的矩阵,1表示这个灯被点击过,0表示没有。
要求这个矩阵能够使得原矩阵的灯全灭。
题目链接 :http://acm.pku.edu.cn/JudgeOnline/problem?id=1222
我的思路很简单 模拟+枚举:
刚开始想的是:枚举所有可能的开关状态,对每个状态计算下最后灯的情况,看是否都熄灭 。可是每种按钮有两种状态,一共有30个开关,那么状态数是2的30次方,会超时。
然后想如何减少枚举数目呢?
如果存在某个局部,一旦这个局部状态被确定后,那么剩余其他部分的状态只能是确定的一种或者不多的n种,那么只需要枚举这个局部状态即可。
然后由题意,题中确实存在这样的一个“局部”:
这个局部就是第一行:
因为第一行的各个开关状态确定的情况下,这些开关作用过后,将导致第一行某些灯是亮的,某些灯是灭的。要熄灭第1行某个亮着的灯(假设位于第i列),那么唯一的办法就是按下第二行第i列的开关(因为第一行的开关已经用过了,而第三行及其后的开关不会影响到第一行)。位了使第一行的灯全部熄灭,第二行的合理开关状态就是唯一的。
本题中是一个5*6的矩阵,我们可以采用2进制数枚举的方法来更高效的处理本题。

#include <memory>  
#include <string>  
#include <cstring>  
#include <iostream>  
using namespace std;  
 //因为开关的状态只有0和1,并且每一行有六位,所以可以用一个char类型的字符来表示每一行的状态 
 //所以所有的状态可以用一个一维的字符数组来表示,通过对字符进行位运算来操作 
char oriLights[6];//存储初始灯的状态  
char lights[6];//存储变化中灯的状态  
char result[6];//存储灯的结果状态  
  
int GetBit(char c,int i)//获取字符c的第i位比特的状态  
{  
    return (c>>i)&1;  
}  
void SetBit(char & c,int i,int v)//把字符c的第i位比特为设置为v  
{  
    if(v)  
    {  
        c =c | (1<<i);//如果v为1,则把1左移i位,然后与c进行或运算  
    }  
    else  
        c =c & ~(1<<i);//如果v为0,则把把1左移i位取反,与c进行与运算  
}  
  
void FlipBit(char &c,int i)  //反转第i位比特 
{  
    c ^=(1<<i);//一个数和1异或就会被反转,和0异或不变  
}  
void OutputResult(int t,char result[])  
{  
    cout << "PUZZLE #" <<t<<endl;     //题目中要求的输出格式
    for(int i=0;i<5;i++)  
    {  
        for( int j=0;j<6 ;j++)  
        {  
            cout<<GetBit(result[i],j);  
            if(j<5)  
                cout<< " ";  
        }  
        cout<<endl;  
    }  
}  
  
int main()  
{  
    int T;//有T组测试数据  
    cin>>T;  
    for(int t=1 ;t<=T; ++t)  
    {  
        for(int i=0;i<5;++i)  
            for(int j = 0;j<6 ;++j)  
            {  
                int s;//s代表数据第i行第j列的数据,0或1  
                cin>>s;  
                SetBit(oriLights[i],j,s);  
            }  
        //开始枚举第一行的所有状态  
        for(int n = 0; n<64;++n)  
        {  
            int switchs = n;//switchs 表示当前行的开关状态  
            memcpy(lights,oriLights,sizeof(oriLights));//第一行的状体变化一次,就把初始状态赋给lights  
            for(int i=0; i< 5 ; ++i)  
            {//开始枚举在第一行已经确定的一种状态下,枚举所有行的状态  
                result[i] = switchs;//把第一行的状态保存到结果状态中  
                for(int j = 0;j< 6 ;++j)  
                {//枚举每一行的每一列的状态  
                    if(GetBit(switchs,j))//如果第i行第j列为1的话,表示按了开关
                    {  
                        if(j>0)  
                            FlipBit(lights[i],j-1);  
                        FlipBit(lights[i],j);  
                        if(j<5)  
                            FlipBit(lights[i],j+1);  
                    }  
                }  
                if( i < 5)//开始处理i+1行的灯  
                {  
                    lights[i+1] ^= switchs;//异或运算  
                }  
                switchs = lights[i];//把下一行的状态赋给switchs至于为什么是这样,如果这一行的灯没关,那么下一行的这个位置就必须是1。  
            }  
            if(lights[4]==0)  
            {  
                OutputResult(t,result);  
                break;  
            }  
        }  
  
    }  
    return 0;  
}  

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值