【LeetCode】杨辉三角(将时间复杂度优化到O(K))

LeetCode118. 杨辉三角

题目描述：

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

例子：

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

解题：

分析：

class Solution {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();
        if (numRows == 0) return res;
        res.add(new ArrayList<Integer>());
        res.get(0).add(1);
        for (int i = 1; i < numRows; i++) {
            List<Integer> row = new ArrayList<>();
            //每一行的第一个元素
            row.add(1);
            for (int j = 1; j < i; j++) {
                //中间元素
                row.add(res.get(i - 1).get(j - 1) + res.get(i - 1).get(j));
            }
            //每一行的最后一个元素
            row.add(1);
            res.add(row);
        }
        return res;
    }
}

LeetCode 119. 杨辉三角 II

题目：

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

例子：

输入: 3
输出: [1,3,3,1]

解题：

调用上面的函数，返回：res.get(rowIndex )即可。

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> res = new ArrayList<>();
        res = generate(rowIndex);
        return res.get(rowIndex);
    }
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();
        res.add(new ArrayList<Integer>());
        res.get(0).add(1);
        for (int i = 1; i <= numRows; i++) {
            List<Integer> row = new ArrayList<>();
            //每一行的第一个元素
            row.add(1);
            for (int j = 1; j < i; j++) {
                //中间元素
                row.add(res.get(i - 1).get(j - 1) + res.get(i - 1).get(j));
            }
            //每一行的最后一个元素
            row.add(1);
            res.add(row);
        }
        return res;
    }
}

优化：将时间复杂度优化到O(K)

思路：参考pwu-gao-qiang大神解题，大神写的是JavaScript的代码，这里给出java版本的代码。

多项式的n次方展开公式联系在一起，(a+b)^n=a^n+[C(n,1)]a^(n-1)*b+C(n,2)a^(n-2)b^2+……+C(n-1,n)ab^(n-1)+b^n通项T(k+1)=C(n,k)a^(n-k)*b^k。

第n行第m个数就是C(m,n),根据公式C(n,m) = n!/((n-m)! * m!)就可以算出来，这里面大部分数都会约掉但是这样会超int，所以要学会化简（所以我用了long）。
就拿33行来说，第0个数是1，第1个数是33/1，第2个数是33/1 * 32/2，第3个数是33/1 * 32/2 * 31/3

代码：提交的时候，打败了100%用户。

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> ans = new ArrayList<>();
        Long temp = 1L;
        ans.add(temp.intValue());
        for (int i = 1; i <= rowIndex; i++) {
            temp *= rowIndex - i + 1;
            temp /= i;
            ans.add(temp.intValue());
        }
        // ans.add(1);
        return ans;
    }
}