回溯法——堡垒问题

描述
城堡是一个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

基本思路:想法同8皇后。不同的是,堡垒问题可以在格子上不放堡垒,而八皇后必须要在一行放一个(以我写的代码说明),所以,堡垒问题必须要在if(canplace())前进行search(m+1),或者之后也行。差别就在这里。代码如下,已经经过测试。

#include <iostream>

using namespace std;

int n, big;
int a[4][4];

void search(int m);
int canplace(int m);
void place(int m);
void takeout(int m);
void checkmax();

int main()
{
    char x;

    cin >> n;

    while( n != 0)
    {
        for(int i = 0; i < n; i++)
        {
            for(int j = 0; j < n; j++)
            {
                cin >> x;
                if(x == 'X')a[i][j] = 1;
                else a[i][j] = 0;
            }
        }

        search(0);
        cout << big << endl;
        big = 0;

        /*for(int i = 0; i < n; i++)
        {
            for(int j = 0; j < n; j++)
            {
                cout << a[i][j];
            }
            cout << endl;
        } */

        cin >> n;
    }


}

void search(int m)
{
    int row;
    int col;
    row = m / n;
    col = m % n;

    if(m >= n*n)
    {
        checkmax();
    }
    else
    {
        search(m+1);
        if(canplace(m))
        {
            place(m);
            search(m + 1);
            takeout(m);
        }
    }
}

int canplace(int m)//因为是逐行搜索,所以只要判断在当前位置上方和左方即可。
{
    int row, col;
    row = m / n;
    col = m % n;

    for(int i = col; i >= 0; i--)
    {
        if(a[row][i] == 2)
        {
            return 0;
            break;
        }
        if(a[row][i] == 1)
        {
            break;
        }
    }

    for(int i = row; i >= 0; i--)
    {
        if(a[i][col] == 2)
        {
            return 0;
            break;
                    }
        if(a[i][col] == 1)
        {
            break;
        }
    }

    if(a[row][col] == 1)return 0;

    return 1;
}

void place(int m)
{
    int row;
    int col;
    row = m / n;
    col = m % n;
    a[row][col] = 2;
}

void takeout(int m)
{
    int row;
    int col;
    row = m / n;
    col = m % n;
    a[row][col] = 0;
}

void checkmax()
{
    int i, j;
    int num = 0;

    for(i = 0; i < n; i++)
        for(j = 0;j < n; j++)
        {
            if(a[i][j] == 2)num++;
        }

    if(num > big)big = num;
}
    原文作者:回溯法
    原文地址: https://blog.csdn.net/sayniceoh/article/details/53136352
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞