0
点赞
收藏
分享

微信扫一扫

面试题 04.12. 求和路径,面试题 05.01. 插入

面试题 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 与 ji <= j,且从 0 位开始计算)。

编写一种方法,使 M 对应的二进制数字插入 N 对应的二进制数字的第 i ~ j 位区域,不足之处用 0 补齐。具体插入过程如图所示。

面试题 04.12. 求和路径,面试题 05.01. 插入_路径和


题目保证从 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;
    }
}







举报

相关推荐

0 条评论