C++基礎算法學習——熄燈問題

有一個由按鈕組成的矩陣, 其中每行有6個按鈕, 共5行
– 每個按鈕的位置上有一盞燈
– 當按下一個按鈕后, 該按鈕以及周圍位置(上邊, 下邊,
左邊, 右邊)的燈都會改變狀態
26熄燈問題 POJ1222
– 如果燈原來是點亮的, 就會被熄滅
– 如果燈原來是熄滅的, 則會被點亮
• 在矩陣角上的按鈕改變3盞燈的狀態
• 在矩陣邊上的按鈕改變4盞燈的狀態
• 其他的按鈕改變5盞燈的狀態
27熄燈問題 POJ1222
 與一盞燈毗鄰的多個按鈕被按下時,一個操作會抵消另一次操
作的結果
 給定矩陣中每盞燈的初始狀態,求一種按按鈕方案,使得所有
的燈都熄滅熄燈問題 POJ1222
 輸入:
– 第一行是一個正整數N, 表示需要解決的案例數
– 每個案例由5行組成, 每一行包括6個數字
– 這些數字以空格隔開, 可以是0或1
– 0 表示燈的初始狀態是熄滅的
– 1 表示燈的初始狀態是點亮的
29熄燈問題 POJ1222
 輸出:
– 對每個案例, 首先輸出一行,
輸出字符串 “PUZZLE #m”, 其中m是該案例的序號
– 接着按照該案例的輸入格式輸出5行
• 1 表示需要把對應的按鈕按下
• 0 表示不需要按對應的按鈕
• 每個數字以一個空格隔開

代碼如下：

#include <iostream>
#include<string.h>
using namespace std;
char Oright[5];
char Light[5];
char Result[5];
void SetChar(char &  t,int j,int s);
void OutPut(char * t);
int GetChar(char  t,int j);
void Flip(char & s,int i);
int main()
{
    int switches;
    for(int i=0;i<5;i++)//一個位一個位的存儲。
    {
        for(int j=0;j<6;j++)
        {
            int s;
            cin>>s;
            SetChar(Oright[i],j,s);
        }
    }
    for(int i=0;i<64;i++)
    {
        memcpy(Light,Oright,sizeof(Oright));
        switches=i;
        for(int j=0;j<5;j++)//對於i的值，每行開始測試
        {
            Result[j]=switches;//Result存儲的是開關的按壓方案。
            for(int k=0;k<6;k++)//每列開始測試
            {
                if(GetChar(switches,k))
                {
                    if(k>0)
                        Flip(Light[j],k-1);//左邊開始變化
                    Flip(Light[j],k);
                    if(k<5)
                        Flip(Light[j],k+1);//右邊開始變化
                }
            }
            if(j<4)
                Light[j+1]^=switches;
            switches=Light[j];
        }
        if(Light[4]==0)
        {
            cout<<"結束！"<<endl;
            OutPut(Result);
            break;
        }
    }
    return 0;
}
void Flip(char & s,int i)
{
    s=s^(1<<i);
}
int GetChar(char  t,int j)
{
    return (t>>j)&1;
}
void SetChar(char & t,int j,int s)
{
    if(s==1)
        t=t|(1<<j);
    else
         t=t&(~(1<<j));
}
void OutPut(char * t)
{
    for(int i=0;i<5;i++)
    {
        for(int j=0;j<6;j++)
        {
            cout<<GetChar(t[i],j);
            cout<<"   ";
        }
        cout<<endl;
    }
}