题目
泰波那契序列 Tn 定义如下:
T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2
给你整数 n
,请返回第 n 个泰波那契数 Tn 的值。
示例 1:
输入:n = 4 输出:4 解释: T_3 = 0 + 1 + 1 = 2 T_4 = 1 + 1 + 2 = 4
示例 2:
输入:n = 25 输出:1389537
提示:
0 <= n <= 37
- 答案保证是一个 32 位整数,即
answer <= 2^31 - 1
。
代码实现
1.创建dp表
class Solution {
public:
int tribonacci(int n)
{
if(n==0) return 0;//处理边界情况
if(n==1||n==2) return 1;
vector<int> dp(n+1);
dp[0]=0;dp[1]=1,dp[2]=1;//初始化数据
for(int i=3;i<=n;i++)//创建dp表
{
dp[i]=dp[i-1]+dp[i-2]+dp[i-3];
}
return dp[n];
}
};
2.滚动数组
class Solution {
public:
int tribonacci(int n)
{
if(n==0) return 0;
if(n==1||n==2) return 1;
int a=0,b=1,c=1,d=0;
for(int i=3;i<=n;i++)
{
d=a+b+c;
a=b;b=c;c=d;
}
return d;
}
};
题目
三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例1:
输入:n = 3 输出:4 说明: 有四种走法
示例2:
输入:n = 5 输出:13
提示:
- n范围在[1, 1000000]之间
代码实现
1.创建dp表
class Solution {
public:
int waysToStep(int n)
{
if(n==1) return 1;
if(n==2) return 2;
if(n==3) return 4;
const int MOD=1e9+7;
vector<int> dp(n+1);
dp[1]=1;dp[2]=2;dp[3]=4;//先初始化前几位数据
for(int i=4;i<=n;i++)
{
dp[i]=((dp[i-1]+dp[i-2])%MOD+dp[i-3])%MOD;//创建dp表
}
return dp[n];
}
};
2.滚动数组
class Solution {
public:
int waysToStep(int n)
{
if(n==1) return 1;
if(n==2) return 2;
if(n==3) return 4;
const int MOD=1e9+7;
int a=1,b=2,c=4,d=0;//初始化数据
for(int i=4;i<=n;i++)
{
d=((a+b)%MOD+c)%MOD;//滚动的朝n走
a=b;
b=c;
c=d;
}
return d;
}
};