HDU 1565 方格取数(1)

链接

思路

　　状压dp，dp[i][j]：表示到第i行，当前状态为s的最大值。预处理在每一行选的时候可能的状态，及这个状态的价值。

　　转移方程：dp[i][j] = max(dp[i][j],dp[i-1][k]+sum[i][j]) k与j不冲突。

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>

using namespace std;

int sta[20010],dp[22][20010],sum[22][20010],a[22][22];
int n,tot;


inline int read() {
    int x = 0,f = 1;char ch = getchar();
    for (; !isdigit(ch); ch=getchar()) if(ch=='-') f=-1;
    for (; isdigit(ch); ch=getchar()) x = x*10+ch-'0';
    return x * f;
}
void init() {
    tot = 0;
    memset(dp,0,sizeof(dp));
    memset(sum,0,sizeof(sum));
    memset(sta,0,sizeof(sta));
    for (int i=1; i<=n; ++i) 
        for (int j=1; j<=n; ++j) 
            a[i][j] = read();

}
void getsta() {
    int t = (1 << n) - 1;
    for (int i=0; i<=t; ++i) {
        if (i & (i << 1) ) continue;
        sta[++tot] = i;
    }
}
void getsum() {
    for (int i=1; i<=n; ++i) {
        for (int j=1; j<=tot; ++j) {
            for (int k=0; k<n; ++k) {
                sum[i][j] += ((1<<k)&sta[j])?a[i][n-k]:0;
            }
        }
    }
}
void getdp() {
    for (int i=1; i<=tot; ++i) dp[1][i] = sum[1][i];
    for (int i=2; i<=n; ++i) {
        for (int j=1; j<=tot; ++j) {
        for (int k=1; k<=tot; ++k) {
                if (sta[j] & sta[k]) continue;
                dp[i][j] = max(dp[i][j],dp[i-1][k]+sum[i][j]);
            }
        }
    }
}
void getans() {
    int ans = 0;
    for (int i=1; i<=tot; ++i) 
        ans = max(dp[n][i],ans);
    cout << ans << "
";
}
int main() {
    while (scanf("%d",&n) != EOF) {
        init();
        getsta();
        getsum();
        getdp();
        getans();
    }
    return 0;
}