poj 1185 炮兵阵地 状态压缩DP

第一道状态压缩DP，只能处理小规模的数据，大规模数据会爆时间，充分利用位运算。

//============================================================================
//
// >     File        :    poj1185.cpp
// >     Author      :    flowertree
// >     Time        :    2015年12月22日
// >     Algorithm   :    状态压缩DP 经典炮兵阵地
//
//============================================================================

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

#define M 11
#define N 105

int dp[N][N][N];
char map[N][M];
int base[N];
int top;
int num[N];
int n, m;
int fire[N];

int maxcmp(int a, int b)
{
	return a > b ? a : b;
}

int CountBit(int i)		//计算一个数的二进制多少位
{
	int sum = 0;
	while(i != 0)
	{
		i = i & (i - 1);
		sum++;
	}
	return sum;
}

void InitMap()		//地图的状态存起来
{
	memset(base, 0, sizeof(base));
	for(int i = 1; i <= n; i++)
	{
		for(int j = 1; j <= m; j++)
		{
			if(map[i][j] == 'H')
				base[i] += (1 << (m - j));
		}
	}
}

void Init()		//压缩炮的所有情况和数量
{
	top = 0;
	memset(fire, 0, sizeof(fire));
	for(int i = 0; i < (1 << m); i++)
	{
		if(((i & (i << 1)) == 0) 
			&& ((i & (i << 2)) == 0))
			{
				fire[++top] = i;
				num[top] = CountBit(i);
			}
	}
}

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
		cin >> map[i] + 1;
	//压缩地图的状态
	InitMap();
	//压缩统计炮的全部情况及数量
	Init();
	memset(dp, -1, sizeof(dp));
	//初始化第一行
	for(int i = 1; i <= top; i++)
	{
		if((base[1] & fire[i]) == 0)
			dp[1][i][1] = num[i];
	}
	//初始化第二行
	for(int i = 1; i <= top; i++)
	{
		if((base[2] & fire[i]) == 0)
		{
			for(int j = 1; j <= top; j++)
			{
				if((dp[1][j][1] >= 0) && ((fire[j] & fire[i]) == 0))
					dp[2][i][j] = maxcmp(dp[2][i][j], dp[1][j][1] + num[i]);
			}
		}
	}
	//动态规划计算后面炮情况
	for(int i = 3; i <= n; i++)
	{
		for(int j = 1; j <= top; j++)
		{
			if((base[i] & fire[j]) == 0)
			{
				for(int k = 1; k <= top; k++)
				{
					for(int t = 1; t <= top; t++)
					{
						if((dp[i - 1][k][t] >= 0)
							&& ((fire[j] & fire[k]) == 0)
							&& ((fire[j] & fire[t]) == 0))
							dp[i][j][k] = maxcmp(dp[i][j][k], dp[i - 1][k][t] + num[j]);
					}
				}
			}
		}
	}
	//查找最大的炮数量
	int maxsum = 0;
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= top; j++)
			for(int k = 1; k <= top; k++)
				maxsum = maxcmp(maxsum, dp[i][j][k]);
	cout << maxsum << endl;
	system("pause");
	return 0;
}