题目描述:
蒜头君觉得白色的墙面好单调,他决定给房间的墙面涂上颜色。他买了 3 种颜料分别是红、黄、蓝,然后把房间的墙壁竖直地划分成 n 个部分,蒜头希望每个相邻的部分颜色不能相同。他想知道一共有多少种给房间上色的方案。
例如,当 n=5 时,下面就是一种合法方案。
|蓝|红|黄|红|黄|
由于墙壁是一个环形,所以下面这个方案就是不合法的。
|蓝|红|黄|红|黄|蓝|
输入格式
一个整数 n,表示房间被划分成多少部分。(1≤n≤50)
输出格式
一个整数,表示给墙壁涂色的合法方案数。
样例输入
4
样例输出
18
解题思路:
注意:dp[i]代表一共有i个部分的墙,一共有几个方案去给墙涂颜色
(1)首先给dp数组初始化,比如有1部分的墙时的方案数,有2个部分的墙的方案数,有3个部分的墙的方案数。dp[1]=3,dp[2]=6,dp[3]=6;
(2)找出递推式:dp[i]=dp[i-1]+2*dp[i-2];
对于(2)的解释如下:
一共有i个部分的墙
①第i-1个部分的墙和第1部分的墙颜色不同,(这个时候第i个部分的墙既不能和第1个部分的墙颜色一样,也不能和第i-1个部分的墙颜色一样),所以第i个部分的墙的颜色已然确定,只有一种,所以dp[i]=dp[i-1];
②第i-1个部分的墙和第1部分的墙颜色相同,(这个时候第i-2个部分的墙不能和第1以及第i-1个部分的墙颜色一样),所以第i个部分的墙的颜色有两种dp[i]=2dp[i-2];【注意:在dii-2个部分的墙的颜色确定的情况下,第i个部分的墙的颜色有两种方案,所以最终的结果是2dp[i-2]】
所以,dp[i]=dp[i-1]+2dp[i-2];
AC代码:
#include <stdlib.h>
#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <math.h>
using namespace std;
long long dp[55];
int main()
{
int i,n;
cin>>n;
// dp[i]:一共有i个部分的墙,一共有几个方案去给墙涂颜色
dp[1]=3;
dp[2]=6;
dp[3]=6;
for(i=4;i<=n;i++)
dp[i]=dp[i-1]+2*dp[i-2];
cout<<dp[n]<<endl;
return 0;
}