状压dp USACO 奶牛混合起来

版权声明：未经本蒟蒻同意，请勿转载本蒻博客 https://blog.csdn.net/wddwjlss/article/details/82774121

有 $N$头奶牛，第i头奶牛的编号是 $S_i$，编号是唯一的。要求相邻奶牛的编号之差均超过 $K$，求有多少种队形是混乱的。

$dp[i][j]$表示状态为 $i$的情况下最后一个牛的编号为 $j$的方案数。转移方程为 $dp[i+(1<<(k-1))][k]+=dp[i][j];$
初值为 $dp[1<<(i-1)][i]=1;$ 即以第只有一个牛的状态并且这只牛就是最后一只的方案数为1。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,s[100100],kk;
ll dp[100000][50],ans;
int main()
{
	scanf("%d%d",&n,&kk);
	for(int i=1;i<=n;++i)
		scanf("%d",&s[i]);
	for(int i=1;i<=n;++i)
		dp[1<<(i-1)][i]=1;
	for(int i=0;i<(1<<n);++i)
		for(int j=1;j<=n;++j)//枚举之前的最后一只牛 
			if(i&(1<<(j-1)))//这只牛是否在状态中 
				for(int k=1;k<=n;++k)//枚举当前的最后一只牛 
					if(!(i&(1<<(k-1)))&&abs(s[k]-s[j])>kk)//当前的牛不在状态中
						dp[i+(1<<(k-1))][k]+=dp[i][j];	
	for(int i=1;i<=n;i++)
        ans+=dp[(1<<n)-1][i];
	cout<<ans;
	return 0;
}