mixup2 混乱的奶牛
题目链接
关键点:
1、根据题目给出的N,可以看出该题极大概率用到状压DP
2、定状态,因为对于相邻的奶牛间,我们有要求让其差值>k,那么对于每一个奶牛的状态,0表示没选,1表示选,则f[i][j],指状态为i,选择了前j头奶牛的最大的方案数
3、状态转移:每次枚举状态,当前选择的奶牛,接下来选择的奶牛,对于当前选择的,我们要让状态里的当前位为1,对于接下来选择的奶牛,我们要让状态里接下来的那位为0,那么设状态为i,当前选择的为j,接下来选择的为k,转移:
f[i|(1<<(k-1))][k] += f[i][j];
4、初始状态,对于只有第i位为1,选了前i位,这时f[1<<(i-1)][i] = 1,方案数为1
5、最终答案,对于选择了所有的奶牛,只选了前i位的所有值相加
6、注意数据范围,答案要开long long
完整代码
# include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, m;
ll f[(1<<20)+20][20];//状态为i,前j头奶牛能摆放的最多的方案数
int a[20];
int main()
{
cin>>n>>m;
for (int i=1; i<=n; i++)
cin>>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++)
{
if (i&(1<<(j-1)))//选前j个
{
for (int k=1; k<=n; k++)//接下来继续选第k个(比较相邻值
{
if (!(i&(1<<(k-1))) && abs(a[k] - a[j])>m)
f[i|(1<<(k-1))][k] += f[i][j];
}
}
}
}
ll ans = 0;
for (int i=1; i<=n; i++)
ans+=f[(1<<n)-1][i];
cout<<ans;
return 0;
}