android杨辉三角算法,【26】杨辉三角 II

题目

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

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

示例:

输入: 3

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

思路

在“杨辉三角”的基础上，算出来目标行的元素，假如list；

因i行只使用i - 1行的元素，所以我们计算出来i行的元素，就可以丢弃之前的list，来降低空间复杂度。

代码

public List getRow(int rowIndex) {

List result = new ArrayList<>();

for (int i = 0; i <= rowIndex; ++i) {

// 中间变量，用完即弃

List 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;

}

结果

执行结果