有 N N N头奶牛,第i头奶牛的编号是 S i S_i Si,编号是唯一的。要求相邻奶牛的编号之差均超过 K K K,求有多少种队形是混乱的。
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]表示状态为
i
i
i的情况下最后一个牛的编号为
j
j
j的方案数。转移方程为
d
p
[
i
+
(
1
<
<
(
k
−
1
)
)
]
[
k
]
+
=
d
p
[
i
]
[
j
]
;
dp[i+(1<<(k-1))][k]+=dp[i][j];
dp[i+(1<<(k−1))][k]+=dp[i][j];
初值为
d
p
[
1
<
<
(
i
−
1
)
]
[
i
]
=
1
;
dp[1<<(i-1)][i]=1;
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;
}