POJ 1321 棋盘问题

题目在http://acm.pku.edu.cn/JudgeOnline/problem?id=1321
steve同学用java写了这道题老是runtime error,我让他把数组开大点,说结果还是runtime error,于是我写了一下,结果ac了,后来他用c++写也ac了。
这个题就是回溯法,和八皇后类似。

/**
*Author fuliang
*poj 1321
*/
#include
#include
using namespace std;

const int MAX = 9;
char board[MAX][MAX];//记录棋盘状态
bool placed_c[MAX];//记录一列是否已经放过棋子
int count;//放棋子的方案数
int num_p;//已放棋子数目
int n,k;//棋盘n*n,放的棋子数k

/*
*是否可以放棋子
*/
bool can_place(int i,int j){
return !placed_c[j] && board[i][j] == '#';
}
/*
*深搜/回溯
*/
void DFS(int i){
if(num_p == k){
count++;
return;
}

if(i >= n)
return;

int j;
for(j = 0; j < n; j++){
if(can_place(i,j)){
placed_c[j] = true;
num_p++;
DFS(i+1);
placed_c[j] = false;
num_p--;
}
}
DFS(i+1);//i行不放棋子
}

int main(){
int i,j;
while(cin >> n >> k, k != -1){
for(i = 0; i < n; i++)
for(j = 0; j < n; j++)
cin >> board[i][j];
count = 0;
num_p = 0;
memset(placed_c,false,sizeof(placed_c));
DFS(0);
cout << count << endl;
}
return 0;
}

你可能感兴趣的:(ACM)