【算法实验二】【回溯法】【1006】堡垒问题

时限：1000ms 内存限制：10000K  总时限：3000ms

描述

城堡是一个4×4的方格，为了保卫城堡，现需要在某些格子里修建一些堡垒。城堡中的某些格子是墙，其余格子都是空格，堡垒只能建在空格里，每个堡垒都可以向上下左右四个方向射击，如果两个堡垒在同一行或同一列，且中间没有墙相隔，则两个堡垒都会把对方打掉。问对于给定的一种状态，最多能够修建几个堡垒。

 

输入

每个测例以一个整数n（1<=n<=4）开始，表示城堡的大小。接下来是n行字符每行n个，‘X’表示该位置是墙，‘.’表示该位置是空格。n等于0标志输入结束。

 

输出

每个测例在单独的一行输出一个整数：最多修建堡垒的个数。

 

输入样例

4
.X..
....
XX..
....
2
XX
.X
3
.X.
X.X
.X.
3
...
.XX
.XX
4
....
....
....
....
0

 

输出样例

5
1
5
2
4

#include<iostream>
#include<vector>

using namespace std;

int search(int m);
bool canplace(int m);

int a[50][50];//-1表示墙壁，1表示炮，0表示未放置
int max_num = 0;
int tmp_max = 0;
int n;//迷宫边长
vector<int> output;

int main() {

	while (cin >> n && n != 0) {
		tmp_max = 0;
		max_num = 0;
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++) {
				char k;
				cin >> k;
				if (k == 'X')
					a[i][j] = -1;
				else
					a[i][j] = 0;
			}
		output.push_back(search(0));
	}
	for (int i = 0; i < output.size(); i++) {
		cout << output.at(i) << endl;
	}
}

int search(int m) {
	int row = m / n;
	int col = m % n;

	if (m == n*n) {
		if (max_num < tmp_max)
			max_num = tmp_max;
		return max_num;
	}

	if (canplace(m)) {
		a[row][col] = 1;//放置
		tmp_max++;
		search(m + 1);

		a[row][col] = 0;//回溯
		tmp_max--;
	}
	search(m + 1);
}

bool canplace(int m) {
	int row = m / n;
	int col = m % n;
	
	if (a[row][col] == -1)//是墙，不能放
		return false;

	//检查列
	for (int i = n - 1; i >= 0; i--){
		if (a[i][col] == -1)//同列有墙
			break;//安全，可以放
		if (a[i][col] == 1)//同列有炮
			return false;//危险，不可放
	}

	//检查行
	for (int i = n - 1; i >= 0; i--) {
		if (a[row][i] == -1)//同列有墙
			break;//安全，可以放
		if (a[row][i] == 1)//同列有炮
			return false;//危险，不可放
	}

	return true;
}