Leetcode - 杨辉三角II E[119]

问题描述

给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中，每个数是它左上方和右上方的数的和。
示例 1:
        输入: rowIndex = 3
        输出: [1,3,3,1]
示例 2:
        输入: rowIndex = 0
        输出: [1]
示例 3:
        输入: rowIndex = 1
        输出: [1,1]
提示:
    0 <= rowIndex <= 33

方法1

        根据杨辉三角的规律求出前 rowindex 行，然后返回需要的行。

代码

        public List<Integer> getRow(int rowIndex) {
            List<List<Integer>> lists = new ArrayList<>();
            for(int i = 0;i <= rowIndex;i++){
                List<Integer> aa = new ArrayList<>();
                for(int j = 0;j <= i;j++){
                    if(j == 0 || j == i){
                        aa.add(1);
                        continue;
                    }
                    aa.add(lists.get(i - 1).get(j - 1) + lists.get(i - 1).get(j));
                }
                lists.add(aa);
            }
            return lists.get(rowIndex);
        }

执行结果

方法2

        也是根据规律求出前 rowindex 行，然后返回需要的行。但是我们只存储杨辉三角的 一半，在返回的时候对数据进行处理然后返回完整的杨辉三角的一行。

        注意：

                1. 在存储杨辉三角的时候，如何设置每行存储终点下标的设置

                2. array[i - 1][j] 不存在

                3. 还原 返回List 集合 开始还原的下标的设置

代码

        public List<Integer> getRow(int rowIndex) {
            List<List<Integer>> lists = new ArrayList<>();
            for (int i = 0; i <= rowIndex; i++) {
                List<Integer> aa = new ArrayList<>();
                for (int j = 0; j <= i / 2; j++) {
                    if (j == 0) {
                        aa.add(1);
                        continue;
                    }
                    if (lists.get(i - 1).size() < j + 1) {
                        aa.add(lists.get(i - 1).get(j - 1) * 2);
                    } else {
                        aa.add(lists.get(i - 1).get(j - 1) + lists.get(i - 1).get(j));
                    }
                }
                lists.add(aa);
            }
            if (rowIndex > 0) {
                for (int i = (rowIndex + 1) / 2 - 1; i >= 0; i--) {
                    lists.get(rowIndex).add(lists.get(rowIndex).get(i));
                }
                return lists.get(rowIndex);
            }
            return lists.get(0);
        }

执行结果

在计算的时候我们可以只使用一个数组

因为第 i 行 只使用了第 i -1 行的数据，所有采用滚动数组的方式对原先代码进行优化

代码：

            List<Integer> list = new ArrayList<>();
            list.add(1);
            for(int i = 1;i <= rowIndex;i++){
                list.add(0);
                for(int j = i;j > 0;j--){
                    list.set(j,list.get(j) + list.get(j - 1));
                }
            }
            return list;

执行结果和上图差不多