POJ-1222EXTENDED LIGHTS OUT-位运算枚举模板

传送门：http://poj.org/problem?id=1222

题意：开关灯问题，一幅开关的灯中，给出一种操作，使得灯全关掉，（操作一个开关，相邻的灯也会改变）

思路：利用位运算枚举第一行；

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std;

char orilights[],lights[],ans[];

int GetBit(char c, int i)
{
    return (c>>i) & ;
}
void SetBit(char &c,int i,int v)
{
    if(v)
    {
        c |= (<<i);
    }
    else
    {
        c &= ~(<<i);
    }
}
void FlipBit(char &c, int i)
{
    c ^= ( << i);
}
void Output(int t)
{
    cout<<"PUZZLE #"<<t<<endl;
    for(int i=; i<; ++i)
    {
        for(int j=; j<; ++j)
        {
            cout<< GetBit(ans[i],j);
            if(j<)cout<<" ";
        }
        cout<<endl;

    }
}
int main(){
    int T;
    cin>>T;
    for(int t = ; t<=T; t++)
    {
        for(int i=; i < ; ++i)
        {
            for(int j = ; j < ; ++j)
            {
                int s;
                cin>>s;
                SetBit(orilights[i], j, s);
            }
        }
        //cout<<1<<endl;
        for(int n=; n<; ++n)
        {
            int switchs = n;
            memcpy(lights,orilights,sizeof(orilights));
            for(int i=; i < ; ++i)
            {
                ans[i] = switchs;
                for(int j = ;j < ; ++j)
                {
                    if(GetBit(switchs,j))
                    {
                        if(j>)FlipBit(lights[i],j-);
                        FlipBit(lights[i],j);
                        if(j<)FlipBit(lights[i],j+);
                    }
                }
                if(i<)lights[i+] ^= switchs;
                switchs = lights[i];
            }
            if(lights[]==)
            {
                Output(t);
                break;
            }

        }
    }
    return ;
}