Leetcode __118. 杨辉三角

问题描述

给定一个非负整数 numRows，生成杨辉三角的前 numRows 行。

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

示例:

输入: 5
输出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

思路

• 第一行是1
• 每行的首尾元素是1
• 从第二行开始，除首尾元素外，该位置j元素的值=上一行的j位置元素+上一行j-1位置元素的值
• 从第二行开始，依次放入首元素1，中间元素（去掉首尾），放入尾元素1

实现

 public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> lists = new ArrayList<>();
        if (numRows <= 0) {
            return lists;
        }
        List<Integer> list1 = new ArrayList<>();
        list1.add(1);
        lists.add(list1);
        for (int i = 2; i <=numRows; i++) {
            List<Integer> list = new ArrayList<>();
            list.add(1);
            List<Integer> pre_list = lists.get(i - 2);
            for (int j = 1; j < i - 1; j++) {
                list.add(pre_list.get(j) + pre_list.get(j - 1));
            }
            list.add(1);
            lists.add(list);
        }
        return lists;
    }

或者

 public List<List<Integer>> generate1(int numRows) {
        List<List<Integer>> lists = new ArrayList<>();
        if (numRows <= 0) {
            return lists;
        }
        List<Integer> pre_list = new ArrayList<>();
        pre_list.add(1);
        lists.add(pre_list);
        for (int i = 2; i <=numRows; i++) {
            List<Integer> list = new ArrayList<>();
            list.add(1);
            for (int j = 1; j < i - 1; j++) {
                list.add(pre_list.get(j) + pre_list.get(j - 1));
            }
            list.add(1);
            lists.add(list);
            pre_list=list;
        }
        return lists;
    }

小结

• 输入numRows<1时，返回空数组，并非null
• 第一种实现 List pre_list = lists.get(i - 2);，行号跟下边差1，上一行，再-1，故-2.