LeetCode Everyday - 在二叉树中分配硬币

notCoriander
88 阅读1分钟

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第8天，点击查看活动详情

在二叉树中分配硬币

给定一个有 N 个结点的二叉树的根结点 root，树中的每个结点上都对应有 node.val 枚硬币，并且总共有 N 枚硬币。

在一次移动中，我们可以选择两个相邻的结点，然后将一枚硬币从其中一个结点移动到另一个结点。(移动可以是从父结点到子结点，或者从子结点移动到父结点。)。

返回使每个结点上只有一枚硬币所需的移动次数。  

示例1:

image.png

输入： [3,0,0]
输出： 2
解释： 从树的根结点开始，我们将一枚硬币移到它的左子结点上，一枚硬币移到它的右子结点上。

示例2:

image.png

输入： [0,3,0]
输出： 3
解释： 从根结点的左子结点开始，我们将两枚硬币移到根结点上 [移动两次]。然后，我们把一枚硬币从根结点移到右子结点上。

示例3:

image.png

输入： [1,0,2]
输出： 2

示例4:

image.png

输入： [1,0,0,null,3]
输出： 4

提示:

  1. 1<= N <= 100
  2. 0 <= node.val <= N

解题思路:

  1. 硬币的移动规则看似很复杂，因为一个节点可能需要移出硬币，也可能移入硬币，还要求移动次数最少
  2. 如果一个节点的硬币个数是 x，无论是移出还是移入，把该节点的硬币个数变成 1 的最少移动次数必然是 abs(x - 1)
  3. 发现这个规律后，开始二叉树的通用解题思路：假想你现在站在某个根节点上，你如何知道把当前这棵子树的所有节点配平所需的最小移动次数？
  4. 假想多余的和缺少的硬币都移动到根节点去配平，当然根节点本身也要配平，所以整棵树配平的移动次数就是 Math.abs(left) + Math.abs(right) + (root.val - 1)

我的答案:

/**
 * @param {TreeNode} root
 * @return {number}
 */
var distributeCoins = function (root) {
  let res = 0;
  /** 定义：输入一棵二叉树，返回这棵二叉树中多出的硬币个数，返回负数代表缺少硬币 */
  const getRest = (root) => {
    if (root == null) return 0;
    let left = getRest(root.left);
    let right = getRest(root.right);
    // 让当前这棵树平衡所需的移动次数
    res += Math.abs(left) + Math.abs(right) + (root.val - 1);
    // 实现函数的定义
    return left + right + (root.val - 1);
  };
  getRest(root);
  return res;
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )

avatar
文章
阅读
粉丝