面试题 04.12. 求和路径
给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
输出:
3
解释:和为 22 的路径有:[5,4,11,2], [5,8,4,5], [4,11,7]
提示:
节点总数 <= 10000
class Solution {
public int pathSum(TreeNode root, int sum) {
// 路径和频率表
Map<Integer, Integer> prefixSums = new HashMap<>();
// 初始化路径和为0的频率为1
prefixSums.put(0, 1);
// 调用辅助函数
return dfs(root, 0, sum, prefixSums);
}
private int dfs(TreeNode node, int currentSum, int targetSum, Map<Integer, Integer> prefixSums) {
if (node == null) {
return 0;
}
// 更新当前路径和
currentSum += node.val;
// 检查路径和
int count = prefixSums.getOrDefault(currentSum - targetSum, 0);
// 更新频率表
prefixSums.put(currentSum, prefixSums.getOrDefault(currentSum, 0) + 1);
// 递归遍历子树
count += dfs(node.left, currentSum, targetSum, prefixSums) + dfs(node.right, currentSum, targetSum, prefixSums);
// 回溯
prefixSums.put(currentSum, prefixSums.get(currentSum) - 1);
return count;
}
}
面试题 05.01. 插入
给定两个整型数字 N
与 M
,以及表示比特位置的 i
与 j
(i <= j
,且从 0 位开始计算)。
编写一种方法,使 M
对应的二进制数字插入 N
对应的二进制数字的第 i ~ j
位区域,不足之处用 0
补齐。具体插入过程如图所示。
题目保证从 i
位到 j
位足以容纳 M
, 例如: M = 10011
,则 i~j
区域至少可容纳 5 位。
示例 1:
输入:N = 1024(10000000000), M = 19(10011), i = 2, j = 6
输出:N = 1100(10001001100)
示例 2:
输入:N = 0, M = 31(11111), i = 0, j = 4
输出:N = 31(11111)
class Solution {
public int insertBits(int N, int M, int i, int j) {
// 创建一个全1的掩码
int allOnes = ~0;
// 创建一个左边j+1位是1,右边是0的掩码
int left = allOnes << (j + 1);
// 创建一个左边全是0,右边i位是1的掩码
int right = (1 << i) - 1;
// 合并left和right,得到一个中间部分是0,两边是1的掩码
int mask = left | right;
// 将M左移i位,使其与N中要替换的部分对齐
M <<= i;
// 清除N中从i到j的位
int clearedN = N & mask;
// 将M插入到N中
int ans = clearedN | M;
return ans;
}
}