bzoj 1231: mixup2 混乱的奶牛

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位范围内的整数.

            看到这个数据范围就感觉是状压了。

       首先考虑状态，如果由dfs想能相对好想一些。如果我们要dfs，那么肯定是现在放的是哪个，每个点选或没选，所以我们的状态就是dp[i][state]表示现在选了第i个，且每一个选或没选用1或0表示的状态为state的方案数。

       那么转移也能从dfs中想到了，就是枚举下一个点放谁(to)，在放这个>k即合法的情况下便有如下的转移方程了。

       dp[to][(state|(1<<to))]+=dp[now][state];

       最后考虑边界条件。便是只选了一个的方案数dp[i][(1<<i)]的值为1了。最后求的便是以每一个结尾的方案数之和。

　　下附AC代码。

#include<iostream>
#include<stdio.h>
#include<string.h>
#define maxn 17
using namespace std;
typedef long long ll;
ll n,k;
ll a[maxn];
ll dp[maxn][(1<<17)];
ll myabs(ll now)
{
	return now<0 ? -now : now;
}
int main()
{
	scanf("%lld%lld",&n,&k);
	for(ll i=0;i<n;i++)
	{
		scanf("%lld",&a[i]);
		dp[i][(1<<i)]=1;
	}
	
	for(ll state=0;state<(1<<n);state++)
	{
		for(ll now=0;now<n;now++)
		if((state&(1<<now)))
		{
			for(ll to=0;to<n;to++)
			if((state&(1<<to))==0 && myabs(a[now]-a[to])>k)
			{
				dp[to][(state|(1<<to))]+=dp[now][state];
			}
		}
	}
	
	ll ans=0;
	for(ll i=0;i<n;i++)
	ans+=(dp[i][(1<<n)-1]);
	printf("%lld\n",ans); 
}