Enum:EXTENDED LIGHTS OUT(POJ 1222)

　　　　　　　　　　　　　　　　


　　　　　　　　　　　　　　　　　　　　亮灯

　　题目大意：有一个5*6的灯组，按一盏灯会让其他上下左右4栈和他自己灯变为原来相反的状态，要怎么按才会把所有的灯都按灭？

　　3279翻版题目，不多说，另外这一题还可以用其他方法，比如DFS,BFS，不过这些都挺慢的，月还有一个特别厉害的方法，高斯消元法，等我以后再来写

　　

#include <iostream>
#include <algorithm>
#include <functional>

using namespace std;

static int map[5][6], store[5][6], flip[5][6];
static int dirx[4] = { -1, 0, 0,  1 };
static int diry[4] = { 0, -1, 0, 0 };

int solve(void);
int get_light(const int, const int);

int main(void)
{
int sum_of_martrix, ans, res;
scanf("%d", &sum_of_martrix);

for (int k = 1; k <= sum_of_martrix;k++)
{
ans = INT_MAX;
for (int i = 0; i < 5; i++)//Read Gragh
for (int j = 0; j < 6; j++)
scanf("%d", &map[i][j]);

for (int i = 0; i < 1 << 6; i++)
{
memset(flip, 0, sizeof(flip));
for (int j = 0; j < 6; j++)
flip[0][5 - j] = (i >> j) & 1;
res = solve();
if (res < ans)
{
ans = res;
memcpy(store, flip, sizeof(flip));
}
}
printf("PUZZLE #%d\n", k);
for (int i = 0; i < 5; i++)
{
for (int j = 0; j < 6; j++)
printf("%d ", store[i][j]);
printf("\n");
}
}
return EXIT_SUCCESS;
}

int solve(void)
{
int ans = 0;
for (int i = 1; i < 5; i++)
for (int j = 0; j < 6; j++)
flip[i][j] = get_light(i - 1, j);//如果这一列的上一行是亮灯，那么必须按下这盏灯

for (int j = 0; j < 6; j++)
{
if (get_light(4, j) == 1)//看最后一行的灯的情况是否全部灯都灭了
return INT_MAX;
}

for (int i = 0; i < 5; i++)
for (int j = 0; j < 6; j++)
ans += flip[i][j];
return ans;
}

int get_light(const int y, const int x)
{
int dx, dy, sum = map[y][x];

for (int i = 0; i < 4; i++)
{
dx = x + dirx[i]; dy = y + diry[i];
if (0 <= dx && dx < 6 && 0 <= dy && dy < 5)
sum += flip[dy][dx];
}
return sum % 2;
}