LeetCode 数组专栏：119. 杨辉三角 II（Java版）

最新推荐文章于 2022-03-14 10:22:51 发布

DCB0122

最新推荐文章于 2022-03-14 10:22:51 发布

阅读量676

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/qq_34823656/article/details/81516922

本文介绍了一种优化算法，用于在O(k)空间复杂度下求解杨辉三角的第k行。通过使用组合数学中的公式C(n-1,m-1)，避免了传统算法中需要存储所有前n-1行数据的问题，从而显著提高了算法的效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。
这里写图片描述

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

示例：

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

进阶：你可以优化你的算法到 O(k) 空间复杂度吗？

解题思路：本题的难点在于时间复杂度要求降至O(k)，我们知道如果是普通算法，那么时间复杂度为O(n²)，因为要计算第n行，必须知道第1-n-1行的所有数据。
那么我们可不可以无需知道前n-1行的数据而直接得出第n行的值呢，于是我在网上搜索了关于杨辉三角的数学知识，其中有一个数学公式便可以做到。

杨辉三角形的第n行的m个数可表示为 C(n-1,m-1).

于是问题就迎刃而解了，下面贴出AC代码：

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> res = new ArrayList<Integer>();
        long nk = 1;
        for (int i = 0; i <= rowIndex; i++) {
            res.add((int)nk);
            nk = nk * (rowIndex - i) / (i + 1);
        }
        return res;
    }
}