POJ-2411 Mondriaan's Dream 状态压缩DP

最新推荐文章于 2025-12-12 16:37:00 发布

转载最新推荐文章于 2025-12-12 16:37:00 发布 · 68 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zhsl/archive/2013/03/01/2937993.html

文章标签：

#c/c++

本文提供了一种高效的解题思路，通过状态压缩和位运算实现快速的状态转移，有效减少了状态数量并简化了算法。此外，文章对比了两种不同的解题方案，并分享了一个仅用20行代码解决问题的简洁实现。

　　题目链接：http://poj.org/problem?id=2411

　　啪啦啪啦敲了80+行，1A。结果看Discuss，别人20行就解决了= =！，果然是我想复杂了。我的状态压缩效果不是很好，貌似很挫，因为状态考虑得太多了，没有类化，用了2bit的空间来表示每个格子的状态即当前放的是横向01，没放00，竖向11。而且状态转移的时候考虑的是从后面来判断前面的状态是否可行，这样的话每行就多记录了些状态（需要记录格子为空的情况）。

　　其实简单的做法就是从前一状态推向后一状态，用0表示当前格子放置了，1表示当前格子放置的是竖向的，而且是向下凸出的。状态转移方程：f[k][i]=sum{f[k-1][j]}(i和j状态需匹配)，这样的话转移的状态就少了很多，而且操作很方便。一般的做法就是先用DFS搜索出status，然后再来判断匹配。其实这里有个很好的技巧，可以避免先用DFS来找出status。我是从Discuss那份20行代码里学的，从状态的每个位开始，然后遍历每个状态，根据当前状态来确定后继状态，知道遍历完，具体看下面：

算法核心： //摘自：http://www.cppblog.com/kill-myself/
利用二进制状态压缩保存后n个格子是否放置，利用位运算可以更高效率地状态转移（在本程序中，第i位二进制保存：后n个格子中，在第i列的格子是否已填）。由于可以由前一个格子状态转移，利用滚动数组节省空间。
具体算法：
1、由于每个格子都要填满，所以穷举每个格子。
2、每个格子的状态可以由前一个格子的状态转移得到
a,如果前一个格子某状态中当前格子已填，直接加在当前格子的相应的状态中。
b,如果前一个格子某状态中当前格子未填，加在竖放的状态中。
c,如果前一个格子某状态中当前格子未填，下一个格子未放，且不是最后一列，加在横放的状态中

　　代码如下：

 1 #include<cstdio>
 2 #include<string.h>
 3 long long f[2][4100],a,b,n,m,k,j,p;
 4 int main(){
 5    while(scanf("%d%d",&n,&m),memset(f,0,sizeof(f)),f[0][0]=p=1,a=n>m?n:m,b=n+m-a){
 6       while(a--)
 7        for(j=0;++j<=b;memset(f[p=1-p],0,sizeof(f[p])))
 8            for(k=(1<<b);--k+1;)
 9               if(k&1<<j-1)
10                  f[p][k&~(1<<j-1)]+=f[1-p][k];
11               else{
12                    f[p][k|1<<j-1]+=f[1-p][k];
13                    if(j<b&&!(k&1<<j))
14                       f[p][k|1<<j]+=f[1-p][k];
15               }
16       printf("%lld\n",f[1-p][0]);
17    }
18 }

Orz一下...........

　　顺便说一下，本题还可用矩阵乘法来做，对于亿量级数据，矩阵+二分可以秒杀。基本方法就是转化成图论来做，找经过n条边的回路。可参考：Matrix67<十个利用矩阵乘法解决的经典题目>

我的搓代码：

 1 //STATUS:C++_AC_1047MS_3592KB
 2 #include<stdio.h>
 3 #include<stdlib.h>
 4 #include<string.h>
 5 #include<math.h>
 6 #include<iostream>
 7 #include<string>
 8 #include<algorithm>
 9 #include<vector>
10 #include<queue>
11 #include<stack>
12 #include<map>
13 using namespace std;
14 #define LL __int64
15 #define pii pair<int,int>
16 #define Max(a,b) ((a)>(b)?(a):(b))
17 #define Min(a,b) ((a)<(b)?(a):(b))
18 #define mem(a,b) memset(a,b,sizeof(a))
19 #define lson l,mid,rt<<1
20 #define rson mid+1,r,rt<<1|1
21 const int N=14010,INF=0x3f3f3f3f,MOD=1999997;
22 const LL LLNF=0x3f3f3f3f3f3f3f3fLL;
23 
24 int sta[N],q[N][90],cou[N];
25 LL f[12][N];
26 int n,m,stacou;
27 
28 void dfs(int cur,int a,int one)
29 {
30     if(cur==m){
31         if(one)f[0][stacou]=1;
32         sta[stacou++]=a;
33         return;
34     }
35     dfs( cur+1,a,one&one);
36     dfs( cur+1,a|(1<<(cur<<1)),0 );
37     if(cur+2<=m)dfs( cur+2,a|(15<<(cur<<1)),one&one );
38     return;
39 }
40 
41 void match()
42 {
43     int i,j,p,ok;
44     for(i=0;i<stacou;i++){
45         for(j=0;j<stacou;j++){
46             for(p=0,ok=1;p<m;p++){
47                 if( (sta[i]&(1<<(p<<1)))==0 && (sta[j]&(1<<(p<<1)))!=0 )continue;
48                 else if( (sta[i]&(3<<(p<<1)))==(1<<(p<<1))
49                    && (sta[j]&(1<<(p<<1)))==0 )continue;
50                 else if( (sta[i]&(3<<(p<<1)))==(3<<(p<<1))
51                     && sta[j]&(3<<(p<<1)) )continue;
52                 else {ok=0;break;}
53             }
54             if(ok)q[i][cou[i]++]=j;
55         }
56     }
57 }
58 
59 int main()
60 {
61  //   freopen("in.txt","r",stdin);
62     int k,i,j,ok;
63     LL ans;
64     while(~scanf("%d%d",&n,&m) && (n||m))
65     {
66         if((n*m)&1){printf("0\n");continue;}
67         if(m>n)n^=m^=n^=m;
68         ans=0;
69         mem(f,0);
70         mem(cou,0);
71         stacou=0;
72         dfs(0,0,1);
73         match();
74         for(k=1;k<n;k++){
75             for(i=0;i<stacou;i++){
76                 for(j=0;j<cou[i];j++)
77                     f[k][i]+=f[k-1][q[i][j]];
78             }
79         }
80         k--;
81         for(i=0;i<stacou;i++){
82             for(j=0,ok=1;j<m;j++){
83                 if( (sta[i]&(1<<(j<<1)))==0){ok=0;break;};
84             }
85             if(ok)ans+=f[k][i];
86         }
87 
88         printf("%I64d\n",ans);
89     }
90     return 0;
91 }

转载于:https://www.cnblogs.com/zhsl/archive/2013/03/01/2937993.html