摆碟子问题(动态规划)

描述：

假设你的餐桌上摆放了n个盘子，每个盘子有且只能放1个菜，相邻的盘子最多连续两个菜品相同。求你在桌子上能够摆放菜的方式（个数）。

 

#include<iostream>
using namespace std;
int n, k, dp[2][200001], ans;
int main()
{
	while (scanf("%d %d", &n, &k) != EOF)  //分为两个种情况 dp[0][n] 1.前n个盘子的摆放总数（而且第n-1个盘子和第n个盘子不相同的） 
										  //                dp[1][n] 2.前n个盘子的摆放总数（而且第n-1个盘子和第n个盘子是相同的）
	{											//          则前n个总的摆放总数就为1情况+2情况 既dp[0][n]+dp[1][n]; 	
		dp[0][1] = k;  // 
		dp[1][1] = 0;  //
		for (int i = 2; i <= n; ++i)
		{
			dp[0][i] =( dp[1][i - 1] + dp[0][i - 1])*(k-1);  //n-1个盘子和第n个盘子不同的： 那就是前n-1个盘子的摆放总数乘以k-1
			dp[1][i] = dp[0][i - 1]; //而且第n-1个盘子和第n个盘子是相同的： 那就是相当于前n-1个盘子的摆放总数（因为第n-1个盘子确定了，第n个就定下来了）
		}
		ans = dp[0][n] + dp[1][n];
		printf("%d\n", ans);
	}
	return 0;
}