POJ 1321 棋盘问题

题目链接：https://cn.vjudge.net/problem/POJ-1321

和八皇后问题不同：当k < n 时，某些行可以不必填

思路：从当前行i的'#'开始搜，标记纵坐标，从i+1行到n-1行搜下一个‘＃’(默认从第0行开始)

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define mem(a,b) memset(a,b,sizeof(a));
using namespace std;
#define INF 0x3f3f3f3f
typedef long long ll;
const int maxn = 50005;
int n,k,vis[10],ans;
char s[10][10];
void dfs(int r,int t) {
    if(t == k) {//当棋子总数为k时，停止搜索，返回上一层
        ans++;
        return ;
    }
    for(int i = r + 1; i < n; i++) {//从r+1行到n-1行搜下一个'#'
        for(int j = 0 ; j < n; j++) {
            if(s[i][j] == '#'&&!vis[j]){
                vis[j] = 1;
                dfs(i,t+1);
                vis[j] = 0;
            }
        }
    }
}
int main()
{


    while(cin >> n >> k) {
        ans = 0;
        if(n == -1 && k == -1)
            break;
        for(int i = 0; i < n; i++) {
            cin >> s[i];
        }
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                if(s[i][j] == '#') {
                    vis[j] = 1;// 标记纵坐标
                    dfs(i,1);
                    vis[j] = 0;
                }
            }
        }
        cout << ans << endl;
    }
    return 0;
}