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);
}