BZOJ 1231: [Usaco2008 Nov]mixup2 混乱的奶牛 状压dp

1231: [Usaco2008 Nov]mixup2 混乱的奶牛

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 1116 Solved: 648
[Submit][Status][Discuss]

Description

混乱的奶牛 [Don Piele, 2007] Farmer John的N(4 <= N <= 16)头奶牛中的每一头都有一个唯一的编号S_i (1 <= S_i <= 25,000). 奶牛为她们的编号感到骄傲, 所以每一头奶牛都把她的编号刻在一个金牌上, 并且把金牌挂在她们宽大的脖子上. 奶牛们对在挤奶的时候被排成一支”混乱”的队伍非常反感. 如果一个队伍里任意两头相邻的奶牛的编号相差超过K (1 <= K <= 3400), 它就被称为是混乱的. 比如说，当N = 6, K = 1时, 1, 3, 5, 2, 6, 4 就是一支”混乱”的队伍, 但是 1, 3, 6, 5, 2, 4 不是(因为5和6只相差1). 那么, 有多少种能够使奶牛排成”混乱”的队伍的方案呢?

Input

• 第 1 行: 用空格隔开的两个整数N和K

• 第 2..N+1 行: 第i+1行包含了一个用来表示第i头奶牛的编号的整数: S_i

Output

第 1 行: 只有一个整数, 表示有多少种能够使奶牛排成”混乱”的队伍的方案. 答案保证是 一个在64位范围内的整数.

Sample Input

4 1

3

4

2

1

Sample Output

2

输出解释:

两种方法分别是:

3 1 4 2

2 4 1 3

HINT

Source

题解：
定义f[i][j]表示以第i头牛结尾，此时选择的状态是j的方案数（i被选）。则再枚举一遍所有的牛，如果把这个牛插到队尾合法，则方案数累计。最后再枚举一遍所有牛都选了但结尾的牛不同的方案，统计答案。
定义初始值：f[i][1<< i-1]=1;

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 16 + 2;
const int M = (1<<16)+5;

int s[N],n,K;
ll dp[N][M];

inline int Abs(int a) {return a<0?(-a):a;}

int main(){
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;++i) scanf("%d",&s[i]);
    int tot=(1<<n)-1;
    for(int i=1;i<=n;++i) dp[i][1<<i-1]=1;
    for(int j=0;j<=tot;++j)
    for(int i=1;i<=n;++i){
        if((j|(1<<i-1))!=j) continue;
        for(int k=1;k<=n;++k){
           if(((1<<(k-1))|j)==j||Abs(s[k]-s[i])<=K) continue;
           dp[k][j|(1<<k-1)]+=dp[i][j];
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++) ans+=dp[i][tot];
    cout<<ans<<endl;
    return 0;
}