刷爆力扣

【26】杨辉三角 II

2021-05-17  本文已影响0人  公孙剑人

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii/

题目

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。

叶子节点 是指没有子节点的节点。

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

思路

  1. 在“杨辉三角”的基础上,算出来目标行的元素,假如list;
  2. 因i行只使用i - 1行的元素,所以我们计算出来i行的元素,就可以丢弃之前的list,来降低空间复杂度。

代码

    public List<Integer> getRow(int rowIndex) {
        List<Integer> result = new ArrayList<>();
        for (int i = 0; i <= rowIndex; ++i) {
            // 中间变量,用完即弃
            List<Integer> current = new ArrayList<>();
            for (int j = 0; j <= i; ++j) {
                if (j == 0 || j == i) {
                    // 两边的元素,固定设置1
                    current.add(1);
                } else {
                    // 当前位的元素,即为上一行i-1和i位置的元素之和
                    current.add(result.get(j - 1) + result.get(j));
                }
            }
            // 只保存当前行list
            result = current;
        }
        return result;
    }

结果

执行结果
上一篇 下一篇

猜你喜欢

热点阅读