hdu 4539 郑厂长系列故事——排兵布阵(状态压缩dp)

来源:互联网 发布:秩为1矩阵的几 编辑:程序博客网 时间:2024/05/21 09:25

参考:http://blog.csdn.net/u013480600/article/details/19758029
参考了下思路,还是经验不足,找不对状态。做的时候传错了个参数,调了半个多小时这里写图片描述
注意边界情况。
虽然思路看的别人的,但是自己写出来代码还是很爽的
做的时候0表示能够放士兵的位置,方便后边处理。
理解都在注释里。

#include <bits/stdc++.h>using namespace std;int dp[101][250][250];int rec[250];int row[101];int n,m;/*1表示我的兵的位置能坐下的地方全都是0如果我的兵全部都能放下则我的1对应的全部都是0则按位与运算结果为0才合法所以座位要01互相取反,使0为合法位置*/int calc(int num){    int ret = 0;    while(num)    {        if(num&1) ret++;        num >>= 1;    }    return ret;}//判断上下两行是否兼容bool check(int up, int down){    for(int i = 0; i < m; ++i)    {        if(i == 0) if((up&1)&((down>>1)&1)) return false;        if(i == m-1) if(((up>>i)&1)&((down>>(i-1))&1)) return false;        if(((up>>i)&1)&((down>>(i-1))&1) || ((up>>i)&1)&((down>>(i+1))&1))            return false;    }    return true;}//判断当前行与上一行的上一行是否兼容bool judge(int now, int pre){    for(int i = 0; i < m; ++i)        if((now>>i)&(pre>>i))            return false;    return true;}int main(){    int x = 1<<10,k=0,t;    for(int i = 0; i < x; ++i)    {        if(i&(i<<2)) continue;        rec[k++] = i;    }    rec[k] = x;    while(scanf("%d %d",&n,&m) != EOF)    {        if(n == 0 || m == 0)        {            printf("0\n");            continue;        }        memset(dp,0,sizeof(dp));        memset(row,0,sizeof(row));        for(int i = 0; i < n; ++i)        {            for(int j = 0; j < m; ++j)            {                scanf("%d",&t);                row[i] = (row[i]<<1)|!t;            }        }        x = 1<<m;        int res = 0;        //单独计数第0行        for(int i = 0; rec[i] < x; ++i)        {            if(!(row[0]&rec[i]))            {                dp[0][i][0] = calc(rec[i]);                res = max(res,dp[0][i][0]);            }        }        if(n == 1)        {            printf("%d\n",res);            continue;        }        //单独计数第一行        for(int i = 0; rec[i] < x; ++i)            if(!(row[1]&rec[i]))//如果第二行能放在第二行                for(int j = 0; rec[j] < x; ++j)                    if(!(row[0]&rec[j]))//如果第一行能成功放在第一行                        //接下来只需要判断两行是否兼容                        if(check(rec[j],rec[i]))                        {                            dp[1][i][j] = calc(rec[i]) + calc(rec[j]);                            res = max(res,dp[1][i][j]);                        }        for(int r = 2; r < n; ++r)        {            for(int i = 0; rec[i] < x; ++i)//i表示当前行            {                if(!(row[r]&rec[i]))                {                    for(int j = 0; rec[j] < x; ++j)//j表示当前行的上一行                    {                        if(!(row[r-1]&rec[j]))//还要表示出来j的上一行                        {                            for(int k = 0; rec[k] < x; ++k)                            {                                if(!(row[r-2]&rec[k]))                                {                                    if(check(rec[j],rec[i]) && judge(rec[i],rec[k]))                                    {                                        dp[r][i][j] = max(dp[r][i][j],dp[r-1][j][k]+calc(rec[i]));                                        res = max(res,dp[r][i][j]);                                    }                                }                            }                        }                    }                }            }//            printf("%d\n",res);        }        printf("%d\n",res);    }    return 0;}
阅读全文
0 0