1. 程式人生 > 實用技巧 >119. 楊輝三角 II

119. 楊輝三角 II

給定一個非負索引 k,其中 k ≤ 33,返回楊輝三角的第 k 行。

在楊輝三角中,每個數是它左上方和右上方的數的和。

示例:

輸入: 3
輸出: [1,3,3,1]

class Solution {
    public List<Integer> getRow(int rowIndex) {

        List<List<Integer>> res = new ArrayList<>();
        // 記錄前一個數組中的值。
        List<Integer> prev = new ArrayList<>();
        for(int i = 0; i < rowIndex + 1; i++) {
            List<Integer> curr = new ArrayList<>();
            for(int j = 0; j <= i; j++) {
                // 開始和結束插入1
                if(j == 0 || j == i) {
                    curr.add(1);
                } else {
                    curr.add(prev.get(j - 1) + prev.get(j));
                }
            }
            prev = curr;
            res.add(curr);
        }
        return res.get(rowIndex);
    }
}



class Solution {
    public List<Integer> getRow(int rowIndex) {

        List<Integer> res = new ArrayList<>();
        long curr = 1;
        for(int i = 0; i <= rowIndex; i++) {
            res.add((int)curr);
            // 公式
            curr = curr * (rowIndex - i)/ (i + 1);
        }
        return res;
    }
}

  

進階:

你可以優化你的演算法到 O(k) 空間複雜度嗎?

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/pascals-triangle-ii
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。