题目链接:http://poj.org/problem?id=1321
题意:给定棋盘,让你在棋盘上放k个棋子。要求这K个棋子的任意两个不能放在棋盘中的同一行或者同一列。
代码:(深搜+剪枝)
#include <iostream>
#include <cstdio>
using namespace std;
char Board[10][10];
int n, k;
int nCnt;
int visit[10][10];
//int dir[4][2] = {{1,1}, {1,-1}, {-1,1}, {-1,-1}};
int judge(int x, int y) {
if(x<0 || x >= n || y < 0 || y >= n) return 0;
if(Board[x][y] == '.') return 0;
int i;
for(i=0; i<n; i++)
if(i != y && visit[x][i] == 1) return 0;
for(i=0; i<n; i++)
if(i != x && visit[i][y] == 1) return 0;
return 1;
}
void Dfs(int x, int y, int num) {
if(num + n - x - 1 < k) return; //剪枝,没有这步,TLE。有了就125ms
if(num == k) {
//printf("(%d,%d),%d\n", x, y, nCnt);
nCnt ++;
return;
}
int i, j;
int new_x, new_y;
for(j=0; x+j<n && num + n - x - 1 >= k; j++) {
new_x = x + j;
for(i=0; i<n; i++) {
new_y = i;
if(visit[new_x][new_y] == 1) continue;
if(judge(new_x, new_y)) {
visit[new_x][new_y] = 1;
//printf("(%d,%d)\n", new_x, new_y);
Dfs(new_x, new_y, num+1);
visit[new_x][new_y] = 0;
}
}
}
}
int main() {
int i, j;
while(scanf("%d%d", &n,&k) != EOF) {
if(n == -1 && k == -1) break;
for(i=0; i<n; i++) scanf("%s", Board[i]);
//printf("%c\n", Board[0][n-1]);
int flag = 0;
nCnt = 0;
memset(visit, 0, sizeof(visit));
for(i=0; i<n; i++) {
for(j=0; j<n; j++)
if(Board[i][j] == '#' && visit[i][j] == 0) {
//printf("Go:%d,%d\n",i,j);
visit[i][j] = 1;
Dfs(i,j,1);
visit[i][j] = 0;
}
}
printf("%d\n", nCnt);
}
}
刚开始状态转移没有想清楚,贡献了几次wa.后来没有剪枝,TLE。看着别人的0ms,感觉自己真是弱爆了。