[POJ 1691]Painting A Board[DFS][排序]

题目链接： [POJ 1691]Painting A Board[DFS][排序]

题意分析：

正方形中的块需要被染色，当且仅当该块上方相邻的所有块都被染色时，可以染色。色块可以连续染色，只要你拿着的是这个颜色的画笔即可，换一次画笔算一次使用，问：最少使用多少次可以使得整个方块被染完？

解题思路：

on[i]存储i块上方必须被染色的块，sta存储目前为止染了什么块（二进制存储）。然后将色块从上到下，从左到右排序，这样，每次dfs我们就不会错失色块。

具体dfs方法就是每次从0~n - 1遍历色块，考虑染色和不染色两种情况，用ans记录当前最优解，再用这个最优解作为下届来剪枝（不过感觉好像用处不大2333）。

个人感受：

着手搞起来，其实也没有想象中的那么恶心啊= =

具体代码如下：

#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<sstream>
#include<stack>
#include<string>
#define ll long long
#define pr(x) cout << #x << " = " << (x) << '\n';
using namespace std;

struct Block {
    int ur, uc, dr, dc, color;
    bool operator < (const Block &t)const {
        if (ur == t.ur) return uc < t.uc;
        return ur < t.ur;
    }
}b[20];

int ans, n;
vector<int> on[20];

bool ok(int sta, int x) {
    if (on[x].size() == 0) return 1;
    for (int i = 0; i < on[x].size(); ++i) {
        if (!(sta & (1 << on[x][i]))) return 0;
    }
    return 1;
}

void dfs(int use, int sta) {
    if (use >= ans) return;
    if (sta == (1 << n) - 1) {
        ans = use;
        return;
    }
    int org = sta;
    for (int i = 0; i < n; ++i) {
        if ((sta & (1 << i))|| !ok(sta, i)) continue;
        sta |= 1 << i;
        for (int j = 0; j < n; ++j) {
            if (!(sta & (1 << j)) && b[j].color == b[i].color && ok(sta, j))
                sta |= 1 << j;
        }
        dfs(use + 1, sta);
        sta = org;
    }
}

int main()
{
    #ifdef LOCAL
    freopen("C:\\Users\\apple\\Desktop\\in.txt", "r", stdin);
    #endif
    int kase; scanf("%d", &kase);
    while (kase --) {
        scanf("%d", &n);
        ans = n;
        for (int i = 0; i < n; ++i) {
            on[i].clear();
            scanf("%d%d%d%d%d", &b[i].ur, &b[i].uc, &b[i].dr, &b[i].dc, &b[i].color);
        }
        sort(b, b + n);

        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < n; ++j) {
                if (i == j) continue;
                if (b[j].dr == b[i].ur && ((b[i].uc <= b[j].uc && b[j].uc <= b[i].dc) || (b[i].uc <= b[j].dc && b[j].dc <= b[i].dc))) {
                    on[i].push_back(j);
                }
            }
        }

        dfs(0, 0);

        printf("%d\n", ans);
    }
    return 0;
}