poj 2441 Arrange the Bulls(状压DP)

我们把畜棚有没有被公牛选的状态用一个01串来表示，0101就表示1,3号畜棚被公牛占用，其余是空的。因为一个畜棚只能被一个公牛占用，所以在转移状态时需要加上畜棚的状态。

定义dp[i][s]  ： 前i头牛已经选择完畜棚，当前畜棚的状态为s的选择方案数目;

dp[i][s] += dp[i-1][s  -  {第 i 头牛选择的畜棚  j  |  (j 不属于s)} ]

利用01背包化二维数组变一维数组的思想，可以将 s 反推，去掉[i]这一维。

剩下的一个坑点就是给出的数字是1 ~ M。在状态压缩中利用的是0 ~ (M-1).所以需要将畜棚编号减1

/***********************************************
 * Author: fisty
 * Created Time: 2015/6/21 20:05:31
 * File Name   : 2441.cpp
 *********************************************** */
#include <iostream>
#include <cstring>
#include <deque>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
#include <string>
#include <vector>
#include <cstdio>
#include <bitset>
#include <algorithm>
using namespace std;
#define Debug(x) cout << #x << " " << x <<endl
#define Memset(x, a) memset(x, a, sizeof(x))
const int INF = 0x3f3f3f3f;
typedef long long LL;
typedef pair<int, int> P;
#define FOR(i, a, b) for(int i = a;i < b; i++)
#define MAX_N 22
int N, M;
int a[MAX_N+1][MAX_N+1];
int n[MAX_N+1];
int dp[1<<MAX_N];
int getcount(int x){
    int sum = 0;
    while(x){
        if(x&1) sum++;
        x >>= 1;
    }
    return sum;
}
int main() {
    //freopen("in.cpp", "r", stdin);
    cin.tie(0);
    ios::sync_with_stdio(false);
    cin >> N >> M;
    Memset(dp, 0);
    FOR(i, 0, N){
        cin >> n[i];
        FOR(j, 0, n[i]){
            cin >> a[i][j];
        }
    }
    for(int i = 0;i < n[0]; i++){
        int j = a[0][i]-1;
        dp[1<<j] = 1;
    }
    for(int i = 1; i < N; i++){
        for(int s = (1<<M)-1; s >= 0; s--){
            if(getcount(s) != i+1) continue;   //必须满足前i头奶牛都进行了选择
            for(int k = 0;k < n[i]; k++){
                int j = a[i][k]-1;
                if(!((1<<j)&s)) continue;
                dp[s] += dp[s^(1<<j)];
            }
        }
    }
    int ans = 0;
    for(int i = 0;i < (1 << M); i++){
        if(getcount(i) == N){
            ans += dp[i];
        }
    }
    cout << ans << endl;
    return 0;
}