hdu 2167 Pebbles （状态压缩dp）

题意：给你个n*n的矩阵让你随便取出几个数和最大  要求如果取出x则它周围八个数都不能再被取出

题目链接：点击打开链接

#include<stdio.h>
#include<string.h>

int dp[20][2000];//存第i行第j个合法状态所能得到的最大值
int state[2000];// 存行内无矛盾的合法状态
int map[20][20];// 存矩阵

int size,limit,n;

void in()// 求出所有合法状态并存到state[]中
{
    int i;
    size=0;
    limit=1<<15;
    for(i=0;i<limit;i++)
    {
        if((i&(i<<1))!=0)continue;
            state[size++]=i;
    }
    state[size]=limit;
}
int ok(int i,int j)// 判断第i个合法状态和第j个合法状态是否冲突
{
    if((state[i]&(state[j]<<1))!=0)return 0;
    if((state[i]&(state[j]>>1))!=0)return 0;
    if((state[i]&state[j])!=0)return 0;
    return 1;
}
int get(int x,int y)// 求出第x行第y个合法状态 所取数字之和
{
    int j,sum=0;
    for(j=0;j<n;j++)// 用j扫描第y个合法状态 找出y状态所取的数
    {
        if((state[y]&(1<<j))!=0)
        {
            sum+=map[x][j];
        }
    }
    return sum;
}
int solve()
{
    memset(dp,0,sizeof(dp));
    int i,j,k;

    for(i=0;state[i]<limit;i++)// 初始化第0行
    {
        dp[0][i]=get(0,i);
    }

    int max;
    for(i=1;i<n;i++)// 从第一行向下dp
    {
        for(j=0;state[j]<limit;j++)// 枚举第i行的状态j
        {
            max=0;
            for(k=0;state[k]<limit;k++)// 枚举前一行的状态k
            {
                if(ok(j,k))// 判断状态j，k是否冲突
                {
                    if(max<dp[i-1][k])max=dp[i-1][k];// 不冲突的话求出与dp[i][j]不冲突的dp[i-1][k]的最大值
                }
            }
            dp[i][j]=max+get(i,j);//当前状态的最大值 为上一行与之不冲突的状态的最大值加上当前行所取数的和
        }
    }

    int ans=0;
    for(i=0;state[i]<limit;i++)//遍历最后一行求答案
    {
        if(dp[n-1][i]>ans)ans=dp[n-1][i];
    }
      return ans;
}

int main()
{
    int i,j;
    in();
    while(scanf("%d",&map[0][0])!=EOF)//输入第一个数字
    {
        n=1;
        char ch;
        while(scanf("%c",&ch),ch!='\n')//输入第一行求n
        {
            scanf("%d",&map[0][n++]);
        }
        for(i=1;i<n;i++)//输入剩余的行
        {
            for(j=0;j<n;j++)
                scanf("%d",&map[i][j]);
        }

        limit=1<<n;
        int ans=solve();
        printf("%d\n",ans);
    }
    return 0;
}