打卡leetcode第二天,坚持每天输出一点点。依然做得是基础题,今天解封了,街上的人明显多了起来,希望明天可以更好,嘿嘿。
题目
杨辉三角 II
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
原题链接:https://leetcode-cn.com/explore/orignial/card/recursion-i/257/recurrence-relation/1203/
解题思路
因为k是非负整数,通过审题可知k从第0层开始,因此第0层含有一个元素【1】,第1层中含有两个元素【1,1】,第2层含有三个元素【1,1+1,1】,可知,当k>2时,该层含有k+1个元素,且第0个元素和第k个元素(为了方便编程,从0号元素开始,如果是从1号元素开始则第1个和第k+1个)均为1,在第k层的第m号元素的值为上一层即k-1层的第m-1号元素和m号元素之和。
因此可以写出公式,当k>1时,用Mk表示第k层的数组,Mk-1表示第k-1层的数组,数组为【1,Mk-1[0]+Mk-1[1],.........Mk-1[n-1]+Mk-1[n],1】。
编程实现
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> vec;
if(rowIndex == 0) {
vec.push_back(1);
return vec;
}
if(rowIndex == 1 ) {
vec.push_back(1);
vec.push_back(1);
return vec;
}
vector<int> last_set = getRow(rowIndex - 1);
for(int i = 0; i <= rowIndex ; i ++){
if(i == 0 || i == rowIndex ) vec.push_back(1);
else{
vec.push_back(last_set[i - 1] + last_set[i]);
}
}
return vec;
}
};
提交记录
当k值过大时,用递归的方法会很占用内存而且速度会变慢,我想获取可以用递推公式去解决,因此我去评论区看了一下各位大神的解法。有提出动态规划的,还有直接打出来36个的常数级大佬,哈哈,这些就不说了。还是决定尝试用公式来解题。
递推公式解题
- 递推公式C(n,i) = n!/(i!*(n-i)!)
- 可以推出第n+1层元素是第n层元素的(n-i)/(i+1)
- 可以用循环写出来
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res;
long cur = 1;
for (int i = 0; i <= rowIndex; i++) {
res.push_back(cur);
cur = cur * (rowIndex-i)/(i+1);
}
return res;
}
};
这时的代码看起来就很简洁了 。同时对比一下运行时间。
可以看到执行用时上大大提高,同时用循环写出来的代码内存消耗也优于前者的递归算法。
今天的打卡成功~叮