当前位置: 首页 > >

LeetCode119 杨辉三角 II

发布时间:

给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。



在杨辉三角中,每个数是它左上方和右上方的数的和。


示例:


输入: 3
输出: [1,3,3,1]
进阶:


你可以优化你的算法到 O(k) 空间复杂度吗?
【点评】用改进版的动态规划,将二维dp改为两个dp数组,cur和last,cur是从上一行last演化而来,注意交替变化,res始终记录当前算得的最新值。如果i还继续循环,会把当前行cur赋给res,last赋给cur做下一次计算用,cur赋给last,表示下一次的上一行是当前行。


import java.util.ArrayList;
public class Solution {
public ArrayList getRow(int rowIndex) {
int[] last = new int[rowIndex +1];
int[] cur = new int[rowIndex + 1];
for (int i = 0; i < rowIndex + 1; i++) {
last[i] = 1;
cur[i] =1;
}
int[] res = cur;
for (int i = 2; i < rowIndex + 1; i++) {
for (int j = 1; j < i; j++) {
cur[j] = last[j] + last[j-1];
}
res = cur;
cur = last;
last = res;
}
ArrayList r = new ArrayList<>();
for (int i = 0; i < rowIndex + 1; i++) {
r.add(res[i]);
}
return r;
}
}



友情链接: year2525网 工作范文网 QS-ISP 138资料网 528200 工作范文网 baothai 表格模版