poj 1185 经典状态压缩dp

 1 #include<iostream>
 2 #include<string>
 3 #include<cstdio>
 4 #include<vector>
 5 #include<queue>
 6 #include<stack>
 7 #include<set>
 8 #include<algorithm>
 9 #include<cstring>
10 #include<stdlib.h>
11 #include<math.h>
12 #include<map>
13 using namespace std;
14 #define pb push_back
15 #define ll long long
16 #define mod 100000000
17 int dp[110][65][65],tmp[110],mmax,sum[1<<12],num[1<<12],n,m;
18 char p[110][20];
19 int judge1(int x){//判断是否有相邻的，或是隔着一个的
20     if(x&(x<<1)||x&(x<<2)) return 0;
21     else return 1;
22 }
23 int getnum(int x){
24     int ans=0;
25     while(x){
26         if(x&1) ans++;
27         x>>=1;
28     }
29     return ans;
30 }
31 void gettmp(){
32     memset(tmp,0,sizeof(tmp));
33     for(int i=1;i<=n;i++)
34         for(int j=0;j<m;j++)
35             if(p[i][j]==‘H‘) tmp[i]+=1<<j;
36 }
37 void getmmax(){//每一行可取的状态
38     mmax=0;
39     for(int i=0;i<(1<<m);i++)
40     if(judge1(i))
41         num[mmax]=i,sum[mmax++]=getnum(i);
42 }
43 int main(){
44    while(cin>>n>>m){
45         for(int i=1;i<=n;i++)
46            cin>>p[i];
47         memset(dp,0,sizeof(dp));
48         gettmp();
49         getmmax();
50         for(int i=0;i<mmax;i++){
51             if(num[i]&tmp[1]) continue;
52             dp[1][i][0]=sum[i];
53         }
54         for(int i=2;i<=n;i++){
55             for(int j=0;j<mmax;j++){//当前取得状态
56                if(num[j]&tmp[i]) continue;
57                  for(int k=0;k<mmax;k++){//前一次取的状态
58                     if(num[k]&tmp[i-1]) continue;
59                     if(num[j]&num[k]) continue;
60                     for(int s=0;s<mmax;s++){//前两次取得状态
61                         if(num[s]&tmp[i-2]) continue;
62                         if((num[k]&num[s])||(num[j]&num[s])) continue;
63                         dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][s]+sum[j]);
64                     }
65                  }
66             }
67         }
68         int ans=0;
69         for(int i=0;i<mmax;i++)
70             for(int j=0;j<mmax;j++)
71                 if(!(num[j]&num[i]))
72             ans=max(ans,dp[n][i][j]);
73         cout<<ans<<endl;
74    }
75 }