面试题 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;
}
}