hdu 2569 DP

最新推荐文章于 2019-07-22 09:55:53 发布

oaeo

最新推荐文章于 2019-07-22 09:55:53 发布

阅读量398

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/chaoojie/article/details/7733324

版权

算法专栏收录该内容

74 篇文章 0 订阅

订阅专栏

1、注意用__int64

2、别少考虑情况

dp[i][j][k] 表示当前第i段颜色为j，且第i-1段的颜色为k时的方法数。然后按照题目要求写状态转移，很好理解

#include "stdio.h"
#include "string.h"

void main(){
	int n, k, i, j;
	__int64 dp[40][3][3], s;  
	freopen("in.txt", "r", stdin);
	scanf("%d", &n);
	while(n--){
		scanf("%d", &k);
		if(k==1){
			printf("3\n");
			continue;
		}
		memset(dp, 0, sizeof(dp));
		dp[2][0][0] = dp[2][1][0] = dp[2][2][0] = 1;
		dp[2][0][1] = dp[2][1][1] = dp[2][2][1] = 1;
		dp[2][0][2] = dp[2][1][2] = dp[2][2][2] = 1;
		for(i=3; i<=k; i++){
			dp[i][0][0] = dp[i-1][0][0] + dp[i-1][0][1] + dp[i-1][0][2];
			dp[i][0][1] = dp[i-1][1][0] + dp[i-1][1][1];
			dp[i][0][2] = dp[i-1][2][0] + dp[i-1][2][2];
			dp[i][1][0] = dp[i-1][0][0] + dp[i-1][0][1];
			dp[i][1][1] = dp[i-1][1][0] + dp[i-1][1][1] + dp[i-1][1][2];
			dp[i][1][2] = dp[i-1][2][1] + dp[i-1][2][2];
			dp[i][2][0] = dp[i-1][0][0] + dp[i-1][0][2];
			dp[i][2][1] = dp[i-1][1][1] + dp[i-1][1][2];
			dp[i][2][2] = dp[i-1][2][0] + dp[i-1][2][1] + dp[i-1][2][2];
		}
		s = 0;
		for(i=0; i<3; i++)
			for(j=0; j<3; j++)
				s += dp[k][i][j];
		printf("%I64d\n", s);
	}
}