来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii/
题目
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。
叶子节点 是指没有子节点的节点。
示例:
输入: 3
输出: [1,3,3,1]
思路
- 在“杨辉三角”的基础上,算出来目标行的元素,假如list;
- 因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;
}
网友评论