POJ 1185 炮兵阵地(状态压缩)

/*
推荐：经典状态压缩

我原来四进制表示，结果果断超时！
解题关键：假设一行所有列均为P，将所有可行状态，存储到数组A[]中，经过计算，状态最多有60种，然后再进行DP。否则很容易超时。
d[i][j][k]：表示第i-1行状态为j，第i行状态为k，最大炮兵数。
如果只是d[i][j]，则无法从前面状态推导，虽然前两行状态可以知道，但会存在冲突。
而d[i][j][k]则可有效避免，因为前两行状态是连在一起的

技巧：①、②处可以这样来优化，我原来全部都是for()循环来解决，结果1000多ms
*/

#include <cstdio>
#include <cstring>
int N, M;
char s[15];
int cnt, A[65], v[65];
int d[2][65][65];
void init()
{
	int i, k;
	int pre;
	cnt = 0;
	for(i = 0; i < (1 << 10); ++ i)
	{
		pre = -5;
		int z = 0;
		for(k = 0; k < 10; ++ k)
		{
			if(i & (1 << k))
			{
				if(k - pre > 2)
					pre = k, z ++;
				else
					break;
			}
		}
		if(k == 10)
			A[cnt] = i, v[cnt ++] = z;
	}
}

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

bool check(int i, int j)
{
	return !(i & j);//②
}

int main()
{
	//freopen("e://data.in", "r", stdin);
	init();
	while(scanf("%d%d", &N, &M) != EOF)
	{
		int i, j, k, l;
		///memset(d, -1, sizeof(d));
		int up = (1 << M);
		for(i = 0; i < N; ++ i)
		{
			scanf("%s", s);
			int z = 0;
			for(j = 0; j < M; ++ j)
			{
				if(s[j] == 'H') z |= (1 << (M - 1 - j));//①
			}
			memset(d[1], -1, sizeof(d[1]));
			for(j = 0; j < cnt && A[j] < up; ++ j)
			{
				if(!check(A[j], z)) continue;
				if(i == 0) d[1][0][j] = v[j];
				else if(i == 1)
				{
					for(k = 0; k < cnt && A[k] < up; ++ k)
					{
						if(d[0][0][k] != -1 && check(A[j], A[k]))
							d[1][k][j] = max(d[1][k][j], d[0][0][k] + v[j]);
					}
				}
				else 
				{
					for(k = 0; k < cnt && A[k] < up; ++ k)
					{
						if(check(A[j], A[k]))
						{
							for(l = 0; l < cnt && A[l] < up; ++ l)
							{
								if(check(A[j], A[l]) && d[0][l][k] != -1)
									d[1][k][j] = max(d[1][k][j], d[0][l][k] + v[j]);
							}
						}
					}
				}
			}
			memcpy(d[0], d[1], sizeof(d[0]));
		}
		int _max = 0;
		for(i = 0; i < cnt && A[i] < up; ++ i)
			for(j = 0; j < cnt && A[j] < up; ++ j)
				if(d[1][i][j] > _max)
					_max = d[1][i][j];
		printf("%d\n", _max);
	}
	return 0;
}