HDU 2442

状态压缩DP ， 和HDU2280极其相似

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <iostream>
 4
 5 using namespace std;
 6 const int N = 105;
 7 int dp[N][1<<6][1<<6] , n , m;
 8
 9 //s 表示当前第i行的状态 , u表示上一行状态 , v表示上上行的状态 , cnt表示到当前位置时总共占据的空间
10 void dfs(int s , int u , int v , int cnt , int k , int i) //由右到左，访问到第i行的第k位数字
11 {
12     if(k >= m){
13         dp[i][u][s] = max(dp[i][u][s] , cnt);
14         return;
15     }
16     if(k > 0){
17         if(!(s & (1<<k)) && !(u & (3<<k-1)) && !(v & (1<<k)))
18             dfs(s | (1<<k) , u | (3<<k-1) , v | (1<<k) , cnt + 4 , k+1 , i);
19         if(!(s & (1<<k)) && !(u & (1<<k)) && !(v & (3<<k-1)))
20             dfs(s | (1<<k) , u | (1<<k) , v | (3<<k-1) , cnt + 4 , k+1 , i);
21     }
22     if(k > 1){
23         if(!(s & (1<<k-1)) && !(u & (7<<k-2)) && !(v & (1<<k-1)))
24             dfs(s | (1<<k-1) , u | (7<<k-2) , v | (1<<k-1) , cnt + 5 , k+1 , i);
25         if(!(s & (1<<k-1)) && !(u & (7<<k-2)))
26             dfs(s | (1<<k-1) , u | (7<<k-2) , v , cnt + 4 , k+1 , i);
27         if(!(s & (1<<k)) && !(u & (7<<k-2)))
28             dfs(s | (1<<k) , u | (7<<k-2) , v , cnt + 4 , k+1 , i);
29     }
30     dfs(s, u , v , cnt , k+1 , i);
31 }
32
33 void dfs2(int s , int u , int cnt , int k)
34 {
35     if(k >= m){
36         dp[2][u][s] = max(dp[2][u][s] , cnt);
37         return;
38     }
39     if(k > 1){
40         if(!(s & (1<<k-1)) && !(u & (7<<k-2)))
41             dfs2(s | (1<<k-1) , u | (7<<k-2) , cnt + 4 , k+2);
42         if(!(s & (1<<k)) && !(u & (7<<k-2)))
43             dfs2(s | (1<<k) , u | (7<<k-2) , cnt + 4 , k+1);
44     }
45     dfs2(s,u,cnt,k+1);
46 }
47 int main()
48 {
49     freopen("a.in","rb",stdin);
50     while(~scanf("%d%d",&n,&m)){
51         memset(dp,-1,sizeof(dp));
52
53         if(n == 1){
54             puts("0");
55             continue;
56         }
57
58         dfs2(0,0,0,0);
59
60         for(int i = 3 ; i<=n ; i++){
61             for(int j = 0 ; j < 1<<m; j++){
62                 for(int t = 0 ; t<1<<m ; t++){
63                     if(dp[i-1][t][j]>=0){
64                         dfs(0,j,t,dp[i-1][t][j],0,i);
65                     }
66                 }
67             }
68         }
69
70         int ans = 0;
71         for(int i = 0 ; i<1<<m ; i++){
72             for(int j = 0 ; j<1<<m ; j++)
73             {
74                 //cout<<"in: "<<i<<" "<<" "<<j<<" "<<dp[n][i][j]<<endl;
75                 ans = max(ans , dp[n][i][j]);
76             }
77         }
78         printf("%d\n" , ans);
79     }
80     return 0;
81 }