[leetCode]119.杨辉三角 II

动态规化

使用一个列表保存杨辉三角每一行的值，初始化第0行为[1],后面的几行都可以由前一行计算得到，最后返回最后一行。

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> dp = new ArrayList<>();
        dp.add(new ArrayList<>());
        dp.get(0).add(1);
        for(int i = 1; i < rowIndex+1; i++){
            List<Integer> row = new ArrayList<>();
            List<Integer> preRow = dp.get(i-1);
            row.add(1);
            for(int j = 1; j < i; j++){
                row.add(preRow.get(j)+preRow.get(j-1));
            }
            row.add(1);
            dp.add(row);
        }
        return dp.get(rowIndex);
    }
}

由于每次只用到前一行这个状态所以可以只使用一个变量preRow保存前一行状态,这样优化了空间复杂度。

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> preRow = new ArrayList<>();
        preRow.add(1);
        for(int i = 1; i < rowIndex+1; i++){
            List<Integer> row = new ArrayList<>();
            row.add(1);
            for(int j = 1; j < i; j++){
                row.add(preRow.get(j)+preRow.get(j-1));
            }
            row.add(1);
            preRow = row;
        }
        return preRow;
    }
}

下面是递归的写法，简单易懂

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> row = new ArrayList<>();
        if(rowIndex == 0){
            row.add(1);
            return row;
        }else{
            List<Integer> preRow = getRow(rowIndex - 1);
            row.add(1);
            for(int i = 1; i < rowIndex; i++){
                row.add(preRow.get(i)+preRow.get(i-1));
            }
            row.add(1);
        }
        return row;
    }
}