题目描述

给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。
在「杨辉三角」中，每个数是它左上方和右上方的数的和。

示例：
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

题目分析

每行第一个和最后一个数字都是1，而中间位置 col 可以用上一行的 col – 1 和 col 加在一起。

Java

public List<List<Integer>> generate(int numRows) {
    List<List<Integer>> ans = new ArrayList<>();
    for (int row = 0; row < numRows; row++) {
        List<Integer> list = new ArrayList<>();
        for (int col = 0; col <= row; col++) {
            if (col == 0 || col == row) {
                list.add(1);
            } else {
                int num = ans.get(row - 1).get(col - 1) + ans.get(row - 1).get(col);
                list.add(num);
            }
        }
        ans.add(list);
    }
    return ans;
}

Kotlin

fun generate(numRows: Int): List<List<Int>> {
    val ans = ArrayList<List<Int>>()
    for (row in 0 until numRows) {
        val list = ArrayList<Int>()
        for (col in 0..row) {
            if (col == 0 || col == row) {
                list.add(1)
            } else {
                val num = ans[row - 1][col - 1] + ans[row - 1][col]
                list.add(num)
            }
        }
        ans.add(list)
    }
    return ans
}