要求:返回类似斐波那契数列数列的数组。只是元素为前三个数之和
解法一:
#include <iostream>
#include <vector>
using namespace std;
std::vector<int> tribonacci(std::vector<int> signature, int n) {
vector<int> result;
if (n <= signature.size()) {
for (int i = 0; i < n; i++) {
result.push_back(signature[i]);
}
return result;
}
int count{ 2 };
while (count < (n - 1)) {
signature.push_back(signature[count] + signature[count - 1] + signature[count - 2]);
count++;
}
result = signature;
return result;
}
优秀解法:
std::vector<int> tribonacci(std::vector<int> result, int n){
std::size_t i=2;
while(++i<n) result.push_back(result[i-1]+result[i-2]+result[i-3]);
result.resize(n);
return result;
}
备注:resize函数的使用:resize(n)相当于在一个已经定义好的vector中安放了n个可以存元素的位置,最多只能装这些数据,多了就无处安放了。
|