算法基础-枚举-编程题＃1： 画家问题

编程题＃1： 画家问题

来源: POJ (Coursera声明：在POJ上完成的习题将不会计入Coursera的最后成绩。)

注意： 总时间限制: 1000ms 内存限制: 65536kB

描述

有一个正方形的墙，由N*N个正方形的砖组成，其中一些砖是白色的，另外一些砖是黄色的。Bob是个画家，想把全部的砖都涂成黄色。但他的画笔不好使。当他用画笔涂画第(i, j)个位置的砖时， 位置(i-1, j)、 (i+1, j)、 (i, j-1)、 (i, j+1)上的砖都会改变颜色。请你帮助Bob计算出最少需要涂画多少块砖，才能使所有砖的颜色都变成黄色。

 

输入

第一行是个整数t(1≤t ≤20)，表示要测试的案例数。然后是t个案例。每个案例的首行是一个整数n (1≤n ≤15)，表示墙的大小。接下来的n行表示墙的初始状态。每一行包含n个字符。第i行的第j个字符表示位于位置(i,j)上的砖的颜色。“w”表示白砖，“y”表示黄砖。

 

输出

每个案例输出一行。如果Bob能够将所有的砖都涂成黄色，则输出最少需要涂画的砖数，否则输出“inf”。

 

样例输入

2
3
yyy
yyy
yyy
5
wwwww
wwwww
wwwww
wwwww
wwwww

 

样例输出

0
15 

#include <iostream>
#include <cstring>

using namespace std;

bool IsTheSolution(int wall[17][17], int (*press)[17][17], int n)
{
	//重置press函数除第一行外的其他部分
	for (int i = 2; i <= n; ++i) {
		for (int j = 0; j <= n; ++j) {
			(*press)[i][j] = 0;
		}
	}
	
	//枚举在第一行确定后的n-1行,推算出最后一行是否符合要求
	for (int i = 2; i <= n; i++)
	{
		//枚举每一行的具体情况
		for (int j = 1; j <= n; j++)
		{
			int temp = (wall[i - 1][j] + (*press)[i - 1][j] + (*press)[i - 1][j + 1] + (*press)[i - 1][j - 1] + (*press)[i - 2][j])%2;
			if (temp == 1) //说明墙已经是黄色的了
			{
				(*press)[i][j] = 0;
			}
			else if (temp == 0) //墙是白色的，还是要涂一下
			{
				(*press)[i][j] = 1;
			}
		}
	}

	//判断最后一行的情况，来返回对错0.
	for (int i = 1; i <= n; i++)
	{
		if ((wall[n][i]+(*press)[n][i-1]+ (*press)[n][i+1]+ (*press)[n-1][i]+ (*press)[n][i])%2==0)
		{
			return false;
		}
	}
	return true;
}

int GetStep(int press[17][17], int n)
{
	int count = 0;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			if (press[i][j] == 1)
				count += 1;
		}
	}
	return count;
}

bool IsFinished(int press[17][17], int n)
{
	for (int i = 1; i <= n; i++)
	{
		if (press[1][i] != 1) {
			//cout << "当前枚举尚未完成" << endl;
			return false;
		}
	}
	//cout << "当前枚举已经完成" << endl;
	return true;
}

//void PrintPress(int Press[][17], int n)
//{
//	for(int i = 1; i <= n; i++)
//	{
//		cout << endl;
//		for(int j = 1; j <= n; j++)
//		{
//			cout << Press[i][j] << " ";
//		}
//	}
//}

int main()
{
	int t;
	cin >> t;
	for (int i = 0; i < t; ++i)
	{
		int n;
		cin >> n;
		int wall[17][17]; //最外层包含一层0围墙，因此行和列都要加2行
		int press[17][17];
		for (int i = 0; i < 17; i++)
		{
			for (int j = 0; j < 17; j++)
			{
				wall[i][j] = 0;
				press[i][j] = 0;
			}
		}

		//输入数据并将数据转化成0和1，0表示白色，1表示黄色
		for (int j = 1; j <= n; j++)
		{
			for (int k = 1; k <= n; k++)
			{
				char temp;
				cin >> temp;
				if (temp == 'y')
					wall[j][k] = 1;
				else if (temp == 'w')
					wall[j][k] = 0;
			}
		}

		int minstep = 1000000;
		bool lineOneComplete = false;
		while (lineOneComplete == false)
		{
			int step = 0;
			//利用二进制枚举
			for (int i = 1; i <= n; i++)
			{
				if (press[1][i] > 1)
				{
					press[1][i] = 0;
					press[1][i + 1] += 1;
				}
			}


			//判断是否是合理的解，是否需要更新最小步数
			if (IsTheSolution(wall, &press, n) == true)
			{
				step = GetStep(press, n);
				//cout << "成功枚举的步数是： " << step << endl;
				if (step < minstep)
				{
					minstep = step;
				}
			}
			else {
			}

			//判断是否枚举完毕
			lineOneComplete = IsFinished(press, n);
			press[1][1]++;
		}

		if (minstep == 1000000) {
			cout << "inf" << endl;
		}
		else {
			cout << minstep << endl;
		}
	}
	return 0;
}