链接: 第 N 个泰波那契数
编写代码
class Solution {
public:
int tribonacci(int n) {
if(n == 0)
{
return 0;
}
else
{
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[i] =dp[i-3] + dp[i -2] + dp[i - 1];
}
return dp[n];
}
};
代码空间优化
一般像这种情况我们可以使用滚动数组的方式来解决空间的问题
class Solution {
public:
int tribonacci(int n) {
if(n == 0)
{
return 0;
}
else
{
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 c;
}
};