HDU4539+状态压缩DP

状态压缩DP

对于某一行的状态可以由前面的两行推出。

即：dp[ i ][ j ][ k ] = max( dp[ i ][ j ][ k ] , dp[ i-1 ][ k ] [ k2 ] + ones[ j ] );

其中i-1表示前1行，k2是前2行的状态。

/*
题意：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 ){
			state[ cnt ] = i;
			ones_state[ cnt ] = Count_ones( i );
			cnt++;
		}
	}
	//printf("cnt=%d\n",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\n",ans);
	}
	return 0;
}