bzoj1231: [Usaco2008 Nov]mixup2 混乱的奶牛
思路:状压dp,设f[i][j]表示当前已经选出的牛的状态为i,最后一头选出的牛为j的方案数。
然后注意就是初值不能是f[0][i]=1,因为所有牛本来都可以第一个被选中,然而这样一定初值有些牛可能就无法被第一个选出了,因此应该是f[(1<<i)][i]=1。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<cmath> using namespace std; #define maxn 17 #define maxs 1<<17 int n,K; int a[maxn]; long long f[maxs][maxn]; int main(){ scanf("%d%d",&n,&K); for (int i=1;i<=n;i++) scanf("%d",&a[i]); for (int i=1;i<=n;i++) f[1<<(i-1)][i]=1; for (int i=0;i<(1<<n);i++) for (int j=1;j<=n;j++) for (int k=1;k<=n;k++) if (!(1<<(k-1)&i)&&abs(a[k]-a[j])>K) f[i|(1<<(k-1))][k]+=f[i][j]; long long ans=0; for (int i=1;i<=n;i++) ans+=f[(1<<n)-1][i]; printf("%lld\n",ans); return 0; }