POJ1222

学到几个新东西

• **利用位操作也许会使问题的处理变得更加方便！
• ** 异或运算符的妙用：
  
  • 1.交换两数更高效
  • 2.反转某一位（参见下面的程序中的Flip函数）

        a ^= b;        b ^= a;         a ^= b;

#include <memory>#include <string>#include <cstring>#include <iostream>using namespace std;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 |= ( 1 << i);    else        c &= ~( 1 << i);}void Flip(char & c, int i){// 将c 的第i 位为取反    c ^= ( 1 << i);}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(){    char oriLights[5]; // 最初灯矩阵，一个比特表示一盏灯    char lights[5]; // 不停变化的灯矩阵    char result[5]; // 结果开关矩阵    char switchs;  // 某一行的开关状态    int T;    cin >> T;    for( int t = 1; t <= T; ++ t)    {        memset(oriLights,0,sizeof(oriLights));        for( int i = 0;i < 5; i ++ )        { // 读入最初灯状态            for( int j = 0; j < 6; j ++ )            {                int s;                cin >> s;                SetBit(oriLights[i],j,s);            }        }        for( int n = 0; n < 64; ++n )        { // 遍历首行开关的64 种状态            memcpy(lights,oriLights,sizeof(oriLights));            switchs = n; // 第i 行的开关状态            for( int i = 0;i < 5; ++i )            {                result[i] = switchs; // 第i 行的开关方案                for( int j = 0; j < 6; ++j )                {                        if( GetBit(switchs,j))                    {                        if( j > 0)                            Flip(lights[i],j-1);// 改左灯                        Flip(lights[i],j);// 改开关位置的灯                        if( j < 5 )                            Flip(lights[i],j+1);// 改右灯                    }                }                if( i < 4 )                    lights[i+1] ^= switchs;// 改下一行的灯                switchs = lights[i]; // 第i+1 行开关方案和第i 行灯情况同            }            if( lights[4] == 0 )            {                OutputResult(t,result);                break;            }        } // for( int n = 0; n < 64; n ++ )    }    return 0;}

“