【NOJ1006】【算法实验二】堡垒问题

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>
 
using namespace std;

char map[4][4];    //城堡
int n;  //城堡边长
int maxn;   //最多城堡个数
int curn;   //当前已修建的城堡个数

void dfs(int m);    //搜索第m个格子，从0开始一行一行编号
bool canbuild(int x, int y);   //map[x][y]这一格能否修建城堡
 
int main()
{
    while(cin>>n&&n)
    {
        //数据输入
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                cin>>map[i][j];
            }
        }
        //初始化
        maxn=curn=0;
        //算法执行
        dfs(0);
        //结果输出
        cout<<maxn<<endl;
    }
    return 0;
}
 
 
void dfs(int m)
{
    if(m==n*n)  //所有格子都搜完
    {
        maxn=max(maxn, curn);
    }
    else
    {
        int x=m/n;  //行
        int y=m%n;  //列
        if(canbuild(x, y))  //判断fort[x][y]这个格子能否修建城堡
        {
            curn++;
            map[x][y]='O'; //在此处修建城堡
            dfs(m+1);
            curn--;
            map[x][y]='.'; //还原
        }
        dfs(m+1);
    }
}
 
bool canbuild(int x, int y)    //判断fort[x][y]这个格子能否修建城堡
{
    if(map[x][y]=='X')     //本格是墙
    {
        return false;
    } 
    for(int i=x-1; i>=0; i--)   //向上搜索同列元素
    {
        if(map[i][y]=='O')     //若同列有城堡则返回false
        {
            return false;
        }
        if(map[i][y]=='X')     //遇到墙则跳出
        {
            break;
        }
    } 
    for(int i=y-1; i>=0; i--)   //向左搜索同行元素
    {
        if(map[x][i]=='O')     //若同行有城堡则返回false
        {
            return false;
        }
        if(map[x][i]=='X')     //遇到墙则跳出
        {
            break;
        }
    }
    return true;
}