領釦LintCode演算法問題答案-1354. 楊輝三角形II

1354. 楊輝三角形II

描述

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

• 注意行下標從 0 開始
• 在楊輝三角中，每個數字是它上面兩個數字的總和。

樣例 1：

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

樣例 2：

輸入: 4
輸出: [1,4,6,4,1]

題解

public class Solution {
    /**
     * @param rowIndex: a non-negative index
     * @return: the kth index row of the Pascal's triangle
     */
    public List<Integer> getRow(int rowIndex) {
        // write your code here
        int[] a = new int[rowIndex + 1];
        a[0] = 1;
        int row = 1;
        while (row++ <= rowIndex) {
            a[row - 1] = 1;
            for (int i = row - 2; i >= 1; i--) {
                a[i] = a[i] + a[i - 1];
            }
        }

        List<Integer> ret = new ArrayList<>();
        for (int n : a) {
            ret.add(n);
        }

        return ret;
    }
}

原題連結點這裡

鳴謝

非常感謝你願意花時間閱讀本文章，本人水平有限，如果有什麼說的不對的地方，請指正。
歡迎各位留言討論，希望小夥伴們都能每天進步一點點。