HDU1045 Fire Net

网吧猪

于 2015-04-24 08:20:52 发布

阅读量386

点赞数

文章标签： dfs hdu

本文链接：https://blog.csdn.net/u013040821/article/details/45241849

版权

大致题意：有障碍阻隔的简单四皇后问题，不需要回溯剪枝。
因为规模非常小，所以直接DFS也不会有任何的超时。我之前大致浏览过这题的代码，不过当时并没有什么思路，于是就干脆“浏览”了，没走心。昨晚在机房搞了接近一个小时，回寝室又搞了许久没解，早上醒来发现测试程序出错导致我以为我的那个judge函数是错的，来回的各种改最后发现还是错的，就从测试程序上找，结果发现写在了每个测试用while n–的外面 orz。
收获：思考一定少对着屏幕而是纸张和书本，不然效率会比较的低下。测试程序测试正确后最好删除而不是注释掉，不然非常乱。

#define LOCAL

#include <iostream>
#include <fstream>

#define MAX_N 4

bool judge(int x, int y);
void DFS(int pos, int cnt);
int input[MAX_N][MAX_N];
//int visited[MAX_N][MAX_N];
int  n, maxn;
int dx[4] = {0, 1, 0, -1};
int dy[4] = {-1, 0, 1, 0};
using namespace std;
bool ok(int x, int y)
{
//  cout << "n: " << n << endl;
    return x >= 0 && x < n && y >= 0 && y < n;
}
int main()
{
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
#endif
    char    in;

    while (cin >> n && n != 0)
    {
        maxn = 0;
        for (int i = 0; i < n; ++i)
        {
            for (int j = 0; j < n; ++j)
            {
                cin >> in;
                if (in == '.')
                    input[i][j] = 0;
                else
                    input[i][j] = 1;
            }
        }
        DFS(0, 0);
        cout << maxn << endl;
    }
    return 0;
}

// 定义0 为街道，1为墙壁，2为碉堡
// 判断是否可以在这个位置(x, y)上装置一个碉堡
// 0 0 0 0
// 0 0 0 0
// 0 0 0 0
// 0 0 0 0

bool judge(int x, int y)
{
    if (input[x][y] != 0)   return false;
    int dx, dy;
    dx = x;
    while (++dx < n && input[dx][y] == 0);
        if (ok(dx, y) && input[dx][y] == 2) return false;
    dx = x;
    while (--dx >= 0 && input[dx][y] == 0);
        if (ok(dx, y) && input[dx][y] == 2) return false;
    dy = y;
    while (++dy < n && input[x][dy] == 0);
        if (ok(x, dy) && input[x][dy] == 2) return false;
    dy = y;
    while (--dy >= 0 && input[x][dy] == 0); 
        if (ok(x, dy) && input[x][dy] == 2) return false;
    return true;
}
// 搜索策略：如果(x, y)可以放置且未访问过，则放置之，cnt++，然后在(x, y)处构造四个
// 方向上的坐标继续访问这四个坐标并且在visited标志位上标志已经访问过了，如果不能放
// 置，则不进入地柜
void DFS(int pos, int cnt)
{
    if (pos == n * n) {
        if (cnt > maxn) {   
            maxn = cnt;
            return;
        }
    } else {
        int tx = pos / n;
        int ty = pos % n;
        if (judge(tx, ty)/* && !visited[tx][ty]*/)
        {
            input[tx][ty] = 2;
    //      visited[tx][ty] = 1;
    //      cout << tx << ' ' << ty << endl;
            DFS(pos + 1, cnt + 1);
            input[tx][ty] = 0;
    //      visited[tx][ty] = 0;
    //      return;
        }
        DFS(pos + 1, cnt);
    }
}