ZOJ-1002 Fire Net

题目链接

ZOJ-1002 Fire Net

题意

给你一个n*n的图（n<=4），‘X’代表墙壁，’ . '代表空地，问你这张图中最多可以放多少炮台（两个炮台不可以放在同行同列，除非中间有墙壁）？

思路

深搜，对于每一个位置，都有放炮台与不放两种选择，递归计算两种选择的最终放置炮台数，不断更新最大值即可。
该题的check函数比较巧妙，具体看代码注释。

#include<iostream>
using namespace std;
int n,ans;
char a[5][5];

bool check(int x,int y)//检查（x，y）能否放置炮台
{
    for(int i=y-1;i>=0;i--)//检查同一行，y之前的位置
    {
        if(a[x][i]=='X')//如果有墙，可以放
            break;
        if(a[x][i]=='@')//如果同一行有炮台
            return 0;
    }
    for(int i=x-1;i>=0;i--)//检查同一列，x之前的位置
    {
        if(a[i][y]=='X')
            break;
        if(a[i][y]=='@')
            return 0;
    }
    return 1;//可以放置炮台
}

void dfs(int m,int tmp)//搜索到第m格时可以放置tmp个炮台
{
    if(m>=n*n)
    {
        ans=max(ans,tmp);//更新最大值
        return;
    }
    else
    {
        int x=m/n;//计算当前所在格子的行数
        int y=m%n;//计算当前所在格子的列数
        if(a[x][y]=='.'&&check(x,y))//如果当前可以放置炮台
        {
            a[x][y]='@';//放置炮台
            dfs(m+1,tmp+1);//搜索下一格
            a[x][y]='.';//返回原状态
        }
        dfs(m+1,tmp);//放与不放都搜索一遍
    }
}

int main()
{
    while(cin>>n&&n)
    {
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                cin>>a[i][j];
        ans=0;
        dfs(0,0);
        cout<<ans<<endl;
    }
    return 0;
}