找实习之从0开始的后端学习日记【9.14】

动规专题地址

118. 杨辉三角

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

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

示例 1:

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

输入: numRows = 1
输出: [[1]]

提示:

1 <= numRows <= 30

动规+滚动数组

class Solution {
    public List<List<Integer>> generate(int numRows) {
        List <List<Integer>> result = new ArrayList<List<Integer>>();
        List<Integer> pre = null;      
        List<Integer> cur = new ArrayList<Integer>();
        for(int i = 1;i<numRows+1;i++){
            pre = cur;
            cur = new ArrayList<Integer>();
            for(int j = 0;j<i;j++){
                if(j==0||j==i-1){
                    cur.add(1);           
                }
                else{
                    cur.add(pre.get(j-1)+pre.get(j));
                }
            }
            result.add(cur);
        }
        return result;

    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

119. 杨辉三角 II

给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。

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

示例 1:

输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:

输入: rowIndex = 0
输出: [1]
示例 3:

输入: rowIndex = 1
输出: [1,1]

提示:

0 <= rowIndex <= 33

数学公式

除了上面的动规以外，再往下优化到O(input)就得用数学公式了。。

要注意一点就是越界问题。。。要用long算！

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> result = new ArrayList<Integer>();
        result.add(1);
        for (int i = 1;i<rowIndex+1;i++){//前半
            result.add(
                new Integer((int)((long)(result.get(i-1))*(long)(rowIndex-i+1)/i))                
                );
        }
        return result;
    }
}
1
2
3
4
5
6
7
8
9
10
11
12