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<Integer> 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<Integer> r = new ArrayList<>();
        for (int i = 0; i < rowIndex + 1; i++) {
            r.add(res[i]);
        }
        return r;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_31617121/article/details/80458425