HDU2167 Pebbles（状压DP）

题目给一张n×n的格子，每个格子都有数字，要从格子中取若干个数字，八个方向相邻的数字不能一起取，问取的数字最大和是多少。

从第一行一行一行看下去，可以发现第1行取哪几列只会影响到第2行，第3行后面的一点影响都没有。即第i行的决策只受i-1行决策的影响。

那么自然想到状态DP——

• dp[i][S]前i行其中第i行取的列的集合是S的取数最大和
• dp[i][S]=max(dp[i-1][S'])+集合S数字和（S是S'的合法的下一行的取法）

虽然题目n最多15，集合S就2¹⁵种状态，但事实上合法的（不能同时取同一行相邻的两列数字）集合S状态只有1000多个。枚举即可转移，为了保证时间复杂度做一些预处理就行了。

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
int read(char *&s){
    int res=-1;
    sscanf(s,"%d",&res);
    while(*s==' ') ++s;
    while(*s>='0' && *s<='9') ++s;
    while(*s==' ') ++s;
    return res;
}
int n,a[15][15],d[15][1<<15],sta[1<<15],sn;
bool isok(int s){
    for(int i=1; i<n; ++i){
        if(((s>>i-1)&1) && ((s>>i)&1)) return 0;
    }
    return 1;
}
bool isok(int x,int y){
    for(int i=0; i<n; ++i){
        if(((x>>i)&1)==0) continue;
        if((y>>i)&1) return 0;
        if(i>0 && ((y>>i-1)&1)) return 0;
        if(i<n-1 && ((y>>i+1)&1)) return 0;
    }
    return 1;
}
int main(){
    char str[1111];
    while(gets(str) && *str){
        n=0;
        char *s=str; int t;
        while(t=read(s),t!=-1) a[0][n++]=t;
        for(int i=1; i<n; ++i){
            gets(str); s=str;
            for(int j=0; j<n; ++j) a[i][j]=read(s);
        }
        sn=0;
        for(int i=0; i<(1<<n); ++i){
            if(isok(i)) sta[sn++]=i; 
        }
        memset(d,0,sizeof(d));
        for(int i=0; i<sn; ++i){
            for(int j=0; j<n; ++j){
                if((sta[i]>>j)&1) d[0][sta[i]]+=a[0][j];
            }
        }
        vector<int> vec[2222];
        for(int i=0; i<sn; ++i){
            for(int j=0; j<sn; ++j){
                if(isok(sta[i],sta[j])) vec[i].push_back(j);
            }
        }
        for(int i=1; i<n; ++i){
            for(int j=0; j<sn; ++j){
                for(int k=0; k!=vec[j].size(); ++k) d[i][sta[j]]=max(d[i][sta[j]],d[i-1][sta[vec[j][k]]]);
                for(int k=0; k<n; ++k){
                    if((sta[j]>>k)&1) d[i][sta[j]]+=a[i][k];
                }
            }
        }
        int res=0;
        for(int i=0; i<sn; ++i){
            res=max(res,d[n-1][sta[i]]);
        }
        printf("%d
",res);
        getchar();
    }
    return 0;
}