HDU4539+状态压缩DP

/*
题意：n行m列的矩阵，1表示可以放东西，0表示不可以。曼哈顿距离为2的两个位置最多只能有一个位置放东西。
问最多放多少个东西。
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
#include<math.h>
#include<map>
using namespace std;
const int maxn = 105;
const int maxm = 12;
const int N = 170;
int mat[ maxn ];
int dp[ maxn ][ N ][ N ];
int state[ N ];
int ones_state[ N ];
int Count_ones( int x ){
    int cnt = 0;
    while( x ){
        if( x&1 )
            cnt++;
        x>>=1;
    }
    return cnt;
}    
int init( int n,int m ){
    memset( state,0,sizeof( state ) );
    memset( ones_state,0,sizeof( state ) );
    int M = 1<<m;
    int cnt = 0;
    for( int i=0;i<M;i++ ){
        if( (i&(i<<2))==0&&(i&(i>>2))==0 ){
            state[ cnt ] = i;
            ones_state[ cnt ] = Count_ones( i );
            cnt++;
        }
    }
    //printf("cnt=%d
",cnt);最多169种状态！！
    return cnt;
}
void DP( int cnt,int n,int m ){
    memset( dp,-1,sizeof( dp ) );
        for( int i=0;i<cnt;i++ ){
            if( (state[i]&mat[0])==0 )
                dp[0][i][0] = ones_state[ i ];
        }//初始化
    for( int i=1;i<n;i++ ){
        for( int j=0;j<cnt;j++ ){
            if( (state[j]&mat[i])==0 ){
                for( int k=0;k<cnt;k++ ){
                    if( (state[j]&(state[k]<<1))==0&&(state[j]&(state[k]>>1))==0 ){
                        for( int k2=0;k2<cnt;k2++ ){
                            if( dp[i-1][k][k2]==-1 ) continue;
                            if( (state[j]&state[k2])==0&&(state[k]&(state[k2]>>1))==0&&(state[k]&(state[k2]<<1))==0 )
                                dp[ i ][ j ][ k ] = max( dp[i][j][k],dp[i-1][k][k2]+ones_state[j] );
                        }
                    }
                }
            }
        }
    }
}
int main(){
    int n,m;
    while( scanf("%d%d",&n,&m)==2 ){
        int cnt = init( n,m );
        memset( mat,0,sizeof( mat ) );
        int tmp;
        for( int i=0;i<n;i++ ){
            for( int j=0;j<m;j++ ){
                scanf("%d",&tmp);
                if( tmp==0 ){
                    mat[ i ] |= (1<<j);
                }
            }
        }
        DP( cnt,n,m );
        int ans = 0;
        for( int i=0;i<cnt;i++ )
            for( int j=0;j<cnt;j++ )
                ans = max( ans,dp[n-1][i][j]);
        printf("%d
",ans);
    }
    return 0;
}