[USACO08NOV]Mixed Up Cows G（状压dp或记忆化dfs）

洛谷链接：[USACO08NOV]Mixed Up Cows G - 洛谷

思路：考虑按顺序放牛，注意到当前牛是否能放只和上一头牛的编号有关（所以能状压，一个数表示已经用过的牛，一个数用来表示最后一头牛）。可以用 f [ i ][ t ]来表示最后一头牛为 i，状态为 t 的可放置情况数。

状压dp代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 17;
int n, k, a[maxn], f[maxn][1<<maxn];
inline int read(){
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
	return x*f;
}
signed main(){
    n=read(); k=read();
    for(int i=0; i<n; i++) a[i]=read(), f[i][1<<i]=1;
    for(int t=1; t<(1<<n); t++) //状态
    for(int i=0; i<n; i++) //cur
    for(int j=0; j<n; j++){ //prev
        if(((1<<i) & t) && ((1<<j) & t) && abs(a[i]-a[j])>k)
            f[i][t] += f[j][t^(1<<i)];
    }
    int ans = 0;
    for(int i=0; i<n; i++) ans += f[i][(1<<n)-1];
    cout<<ans;
}

---

记忆化dfs代码，实际上思路是一样的：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 17;
int n, k, a[maxn], f[maxn][1<<maxn];
inline int read(){
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
	return x*f;
}
int dfs(int step, int t, int cur){
    if(step==n) return 1; //搜到末尾
    if(f[cur][t]) return f[cur][t]; //记忆化
    int res = 0;
    for(int i=1; i<=n; i++){
        if(t & (1<<(n-i))) continue; //已经有了，跳过
        if(cur==0 || abs(a[cur]-a[i]) > k) res += dfs(step+1, t|(1<<(n-i)), i);
    }
    return f[cur][t] = res;
}
signed main(){
    n=read(); k=read();
    for(int i=1; i<=n; i++) a[i]=read();
    cout << dfs(0,0,0);
}