UVA 639 Don't Get Rooked

UVA 639 Don’t Get Rooked

题目大意：类似八皇后的问题，在给定区域放车，车可以横着走也可以竖着走但不能斜着，X是障碍物，求出区域最多能放多少车是它们吃不到对方

解题思路：回溯讨论情况，遍历棋盘放棋子，判断是否会被吃

#include <stdio.h>
#include <string.h>
#include <iostream>
using namespace std;
char map[5][5];
int flag[5][5];
int dir[4][2] = {{0,1},{0,-1},{1,0},{-1,0}};  
int n;
int s;
int jud(int a, int b) {
    int x, y;
    for(int i = 0; i < 4; i++) {
        x = a + dir[i][0];
        y = b + dir[i][1];
        while(x >= 0 && x < n && y >= 0 && y < n) {
            if(map[x][y] == 'X')
                break;
            if(flag[x][y])
                return 0;
            x = x + dir[i][0];
            y = y + dir[i][1];
        }
    }
    return 1;
}
void dfs(int a, int b, int sum) {
    if(sum > s)
        s = sum;
    for(int i = a; i < n; i++)
        for(int j = 0; j < n; j++) {
            if(map[i][j] == '.' && flag[i][j] == 0 && jud(i, j)) {
                flag[i][j] = 1;
                dfs(i, j, sum+1);
                flag[i][j] = 0;
            }
        }
}
int main() {
    while(scanf("%d", &n) && n) {
        memset(flag, 0, sizeof(flag));
        memset(map, 0, sizeof(map));
        for(int i = 0; i < n; i++)
            scanf("%s", map[i]);
        s = 0;
        dfs(0, 0, 0);
        cout << s << endl;

    }
    return 0;

}