状压dp学习

P1896 [SCOI2005]互不侵犯

dp[i][j][sta] 表示第i行状态为sta，一共放置了j个国王

预处理出每一行的合法状态（任意两个国王不相邻 (sta & sta << 1)== 0），并记录每个状态对应的国王数。位运算记得用括号 & 优先级比 == 小。

if sta1和sta2 间任意两个国王能不能相互攻击，那么第i行dp[i][j][sta1]可以由第i - 1行dp[i-1][j+cnt[sta2]][sta2]转移得到。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = (1 << 9);
int n, m, sta;
int cnt[N];
bool f[N];
bool vis[N][N];
ll dp[10][100][N];
void pika(){
    for(int i = 0; i < sta; i++){
        if((i & (i>>1)) == 0){
            f[i] = 1;
            for(int j = i; j ; j >>= 1){
                if(j&1) cnt[i]++;
            }
        }
    }
    for(int i = 0; i < sta; i++){
        if(f[i] == 0) continue;
        for(int j = 0; j < sta; j++){
            if(f[j] == 0) continue;
            if(!(i & j) && !(i & (j<<1)) && !((i<<1) & j)){
                vis[i][j] = vis[j][i] = 1;
            }
        }
    }
}

int main(){
    scanf("%d%d", &n, &m);
    sta = (1 << n);
    pika();
    dp[0][0][0] = 1;
    for(int i = 1; i <= n; i++){
        for(int j = 0; j < sta; j++){
            if(f[j] == 0) continue;
            for(int k = 0; k < sta; k++){
                if(f[k] == 0) continue;
                if(vis[j][k]){
                    for(int l = cnt[j]; l + cnt[k] <= m; l++){
                        dp[i][l + cnt[k]][k] += dp[i-1][l][j];
                    }
                }
            }
        }
    }
    ll ans = 0;
    for(int i = 0; i < sta; i++){
        ans += dp[n][m][i];
    }
    printf("%lld\n", ans);
    return 0;
}

 

 

P1879 [USACO06NOV]玉米田Corn Fields

这题没有限制草地数量，所以只用二维，dp[i][sta] 表示第i行状态为sta

但是这题限制只能在肥沃的草地种植草地，所以sta必须是每行肥沃土地状态的子集。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9;
int n, m;
int f[20];
bool vis[1<<12];
ll dp[20][1<<12];
int main(){
	scanf("%d%d", &n, &m);
	int a;
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			scanf("%d", &a);
			f[i] = (f[i]<<1) + a;
		}
	}
	int sta = 1 << m;
	for(int i = 0; i < sta; i++){
		if((i & i << 1) == 0){
			vis[i] = 1;
		}
	}
	dp[0][0] = 1;
	for(int i = 1; i <= n; i++){
		for(int j = 0; j < sta; j++){
			if(vis[j] == 0 || (j&f[i-1]) != j) continue;
			for(int k = 0; k < sta; k++){
				if(vis[k] == 0 || (k&f[i]) != k) continue;
				if((j & k) == 0) dp[i][k] = (dp[i][k] + dp[i-1][j]) %mod;
			}
		}
	}
	ll ans = 0;
	for(int i = 0; i < sta; i++){
		if(vis[i] == 0 || (i&f[n]) != i) continue;
		ans = (ans + dp[n][i]) % mod;
	}
	printf("%lld\n", ans);
	return 0;
}

P2704 [NOI2001]炮兵阵地

这个题稍微难一点点，但是与上面还是同一类问题。

确定每一行合法状态和前面差不多，每门炮间间距至少为2，当!(sta & sta<<1) && !(sta & sta<<2)时，状态合法。

因为每一行状态与上两行状态有关，所以需要用三维dp

dp[i][sta1][sta2]表示第i行状态为sta1，第i - 1 行状态为是sta2

如果sta1和sta2、sta1和sta3、是sta2和sta3不冲突时，  dp[i][sta1][sta2] 可以由 dp[i-1][sta2][sta3]转移过来

但是这样会爆空间，用01滚动就好了，只记录三行dp值

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 110;
int n, m;
char s[20];
int f[N], cnt[1<<10];
bool vis[1<<10];
ll dp[3][1<<10][1<<10];

int getnum(int x){
	int res = 0;
	while(x){
		if(x & 1) res++;
		x >>= 1;
	}
	return res;
}
int main(){
	scanf("%d%d", &n, &m);
	int sta = 1<<m;
	for(int i = 1; i <= n; i++){
		scanf("%s", s);
		for(int j = 0; j < m; j++){
			f[i] <<= 1;
			if(s[j] == 'P') f[i]++;
		}
	}
	for(int i = 0; i < sta; i++){
		if(!(i & i<<1) && !(i & i<<2)){
			vis[i] = 1;
			cnt[i] = getnum(i);
		}
	}
	for(int i = 0; i < sta; i++){
		if(vis[i] == 0) continue;
		for(int j = 0; j < sta; j++){
			if(vis[j] == 0) continue;
			if((f[1] & j) == j) dp[1][j][i] = cnt[j];
		}
	}
	for(int i = 0; i < m; i++)
		f[0] = (f[0] << 1) + 1;
	for(int i = 2; i <= n; i++){
		for(int j = 0; j < sta; j++){
			if(!vis[j] || (j & f[i-2]) != j) continue;
			for(int k = 0; k < sta; k++){
				if(!vis[k] || (k & f[i-1]) != k || (j & k)) continue;
				for(int l = 0; l < sta; l++){
					if(!vis[l] || (l & f[i]) != l || (k & l) || (l & j)) continue;
					dp[i%3][l][k] = max(dp[i%3][l][k], dp[(i-1)%3][k][j] + cnt[l]);
					//cout << i << " " << dp[i%3][l][k] << endl;
				}
			}
		}
	}
	ll ans = 0;
	for(int i = 0; i < sta; i++){
		for(int j = 0; j < sta; j++){
			ans = max(dp[n%3][i][j], ans);
		}
	}
	printf("%lld\n", ans);
	return 0;
}

P1171 售货员的难题

dp[i][sta] 表示现在在第i 个点，状态为sta, 

dp[i][sta]可以由任意一个点转移过来，该点状态为dp[k][sta^(1<<i)]（sta中包含 k 点），对于每一个状态，必须先得到比该状态少一个点的状态，所以第一重循环需要遍历状态，这样才不会漏解。

#include<bits/stdc++.h>
using namespace std;
const int N = 1<<20;
typedef long long ll;
const int INF = 2e9;
int n;
int a[25][25];
int dp[21][N];

int main(){
	scanf("%d", &n);
	int sta = 1 << n;
	for(int i = 0; i < n; i++){
		for(int j = 0; j < n; j++){
			scanf("%d", &a[i][j]);
		}
	}
	for(int i = 0; i < n; i++){
		for(int j = 0; j < sta; j++){
			dp[i][j] = INF;
		}
	}
	dp[0][1] = 0;
	for(int j = 1; j < sta; j += 2){
		for(int i = 0; i < n; i++){
			if((1<<i & j) == 0) continue;
			for(int k = 0; k < n; k++){
				if(k == i || (j & (1<<k)) == 0) continue;
				dp[i][j] = min(dp[i][j], dp[k][j^(1<<i)] + a[k][i]);
			}
		}
	}
	int ans = INF;
	for(int i = 0; i < n; i++){
		ans = min(ans, dp[i][sta-1] + a[i][0]);
	}
	printf("%d\n", ans);
	return 0;
}