leetcode —— 979. 在二叉树中分配硬币

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

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

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

示例 1：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/distribute-coins-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
————————————————
解题思路：使用深度优先遍历，创建一个函数funs(root)，该函数的作用是计算当前结点root的金币量（这时候当前节点的子节点的金币数都为1）。

当知道当前节点的左子节点的金币数量和当前节点的右子节点的金币数量，就可以知道当左子节点和右子节点的金币数为1时当前节点的金币个数和需要移动的次数。

用变量ans保存整体需要移动的金币次数。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = Noneclass Solution:def __init__(self):self.ans = 0def distributeCoins(self, root: TreeNode) -> int:def funs(root):if not root:return 0elif root.right and root.left:  # 如果存在左子树和右子树numl = funs(root.left) # 计算左子节点的金币数numr = funs(root.right) # 计算右子节点的禁金币数self.ans += abs(numl-1) + abs(numr-1) # 左子节点和右子节点的金币数为1需要移动的次数return root.val + numl + numr - 2 # 更新当前节点的金币数elif root.left: # 如果右子树为空num = funs(root.left) # 计算左子节点的金币数self.ans += abs(num-1)  # 左子节点金币数为1需要移动的次数return root.val + num - 1 # 更新当前节点的金币数elif root.right: # 如果左子树为空num = funs(root.right)  # 计算右子节点的金币数self.ans += abs(num-1)  # 右子节点金币数为1需要移动的次数return root.val + num - 1  # 更新当前节点的金币数else:return root.val  # 如果没有左右子树，返回当前结点的金币数funs(root) return self.ans

该算法会遍历二叉树的每一个节点，因此时间复杂度为O(n)，需要使用堆栈，因此空间复杂度为O(H)，H为二叉树的高度。

leetcode —— 979. 在二叉树中分配硬币相关推荐

LeetCode 979. 在二叉树中分配硬币（DFS）
文章目录 1. 题目 2. DFS 解题 1. 题目给定一个有 N 个结点的二叉树的根结点 root,树中的每个结点上都对应有 node.val 枚硬币,并且总共有 N 枚硬币. 在一次移动中,我们 ...
LeetCode979. 在二叉树中分配硬币
问题:979. 在二叉树中分配硬币给定一个有 N 个结点的二叉树的根结点 root,树中的每个结点上都对应有 node.val 枚硬币,并且总共有 N 枚硬币. 在一次移动中,我们可以选择两个相邻的 ...
leetcode979. 在二叉树中分配硬币（dfs）
给定一个有 N 个结点的二叉树的根结点 root,树中的每个结点上都对应有 node.val 枚硬币,并且总共有 N 枚硬币. 在一次移动中,我们可以选择两个相邻的结点,然后将一枚硬币从其中一个结点移 ...
LeetCode实战：二叉树中的最大路径和
背景为什么你要加入一个技术团队? 如何加入 LSGO 软件技术团队? 我是如何组织"算法刻意练习活动"的? 为什么要求团队的学生们写技术Blog 题目英文 Given a non ...
leetcode算法题--二叉树中的最长交错路径★
原题链接:https://leetcode-cn.com/problems/longest-zigzag-path-in-a-binary-tree/ 嵌套递归(超时) 相关题目:二叉树中的列表 in ...
leetcode算法题--二叉树中序遍历迭代法
原题链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/ 二叉树中序遍历迭代法,栈实现 vector<int> ...
Java实现 LeetCode 623 在二叉树中增加一行（遍历树）
623. 在二叉树中增加一行给定一个二叉树,根节点为第1层,深度为 1.在其第 d 层追加一行值为 v 的节点. 添加规则:给定一个深度值 d (正整数),针对深度为 d-1 层的每一非空节点 N, ...
LeetCode 1602. 找到二叉树中最近的右侧节点（BFS）
文章目录 1. 题目 2. 解题 1. 题目给定一棵二叉树的根节点 root 和树中的一个节点 u ,返回与 u 所在层中距离最近的右侧节点,当 u 是所在层中最右侧的节点,返回 null . 示例 ...
LeetCode 623. 在二叉树中增加一行（BFS/DFS）
文章目录 1. 题目 2. 解题 2.1 BFS 2.2 DFS 1. 题目给定一个二叉树,根节点为第1层,深度为 1.在其第 d 层追加一行值为 v 的节点. 添加规则:给定一个深度值 d (正整 ...

leetcode —— 979. 在二叉树中分配硬币

leetcode —— 979. 在二叉树中分配硬币相关推荐

最新文章

热门文章