POJ 2441 Arrange the Bulls 状态压缩DP

一、题目大意

要求将N头牛放在M个牛棚，每个牛棚只能放一头牛，且每头牛都对牛棚有有所挑剔

例如第一头牛，可以选择的牛棚为一号、二号或四号，则不能将它分配到三号牛棚。

题目要求计算出将所有牛，满足它们的挑剔的情况下，全部分配到牛棚的方案数量。

（1<=N,M<=20）

二、解题思路

我们定义DP数组，DP[S][i] 代表使用集合S里的牛棚，将第 i 头到最后一头牛都分配到牛棚里的方案数，则DP[全集][0]就是最终的答案。

我们定义 i 可以从 N-1到0循环，每层循环里遍历全部的集合S，然后找出 S 与 第 i 头牛喜欢的牛棚的交集，循环交集里的元素 j，进行如下递推式。

dp[S][i]+=dp[ S去掉元素j ][i+1]

对于最后一头牛而言，dp[S][N-1]的值就等于S与它喜欢的牛棚的交集大小，所以可以把最后数组第N+1行初始化为1。

本题目可以定义滑动数组，可读性更佳。

（PS：状态压缩DP需要一点点的二进制入门知识，二进制表示集合的基本操作）

三、代码

#include <iostream>
using namespace std;
const int MAX_N = 20;
const int MAX_M = 20;
int dp[2][1 << MAX_M];
int barns[MAX_N];
int N, M, P;
void solve()
{
    int *crt = dp[0], *next = dp[1];
    fill(crt, crt + (1 << MAX_M), 1);
    for (int i = N - 1; i >= 0; i--)
    {
        for (int S = 0; S < (1 << M); S++)
        {
            next[S] = 0;
            int common = S & barns[i];
            for (int j = common; j > 0; j = j - (j & (-j)))
            {
                next[S] += crt[S & ~(j & (-j))];
            }
        }
        swap(next, crt);
    }
    printf("%d\n", crt[(1 << M) - 1]);
}
int main()
{
    scanf("%d%d", &N, &M);
    int barn;
    for (int i = 0; i < N; i++)
    {
        scanf("%d", &P);
        for (int j = 0; j < P; j++)
        {
            scanf("%d", &barn);
            barns[i] = barns[i] | (1 << (barn - 1));
        }
    }
    solve();
    return 0;
}

四、相关文献

《挑战程序设计竞赛（第2版）》P156-P158 集合的二进制表示

《挑战程序设计竞赛（第2版）》P191-P199 状态压缩DP