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

  题目链接: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 }

 

你可能感兴趣的:(poj)