979在二叉树中分配硬币

灵神真的强,这道题不知道如何遍历,没有什么其他思路,灵神给的边的计数器和移动次数联系在一起实在太妙了

题目

给你一个有 n 个结点的二叉树的根结点 root ,其中树中每个结点 node 都对应有 node.val 枚硬币。整棵树上一共有 n 枚硬币。

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

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

示例 1:

ini 复制代码
输入: root = [3,0,0]
输出: 2
解释: 一枚硬币从根结点移动到左子结点,一枚硬币从根结点移动到右子结点。

代码与解析

使用了深度优先搜索来计算每个节点的多余硬币数目,对于每个节点,计算其左右子树的多余硬币数目,然后再加上当前节点的硬币数目减1(即缺少的硬币数)。累加每个节点移动的次数,并返回最小移动次数。

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int ans = 0; // 记录移动次数

    /**
     * 计算移动硬币的最小次数
     *
     * @param root 树的根节点
     * @return 返回最小移动次数
     */
    public int distributeCoins(TreeNode root) {
        dfs(root); // 调用深度优先搜索函数
        return ans; // 返回最小移动次数
    }

    /**
     * 深度优先搜索计算节点移动情况
     *
     * @param node 当前节点
     * @return 返回节点的多余硬币数目
     */
    public int dfs(TreeNode node) {
        if (node == null) return 0; // 如果节点为空,返回0
        // 计算左右子树的多余硬币数目,加上当前节点的硬币数目减1(即缺少的硬币数)
        int d = dfs(node.left) + dfs(node.right) + node.val - 1;
        ans += Math.abs(d); // 更新移动次数,累加当前节点的移动次数
        return d; // 返回当前节点的多余硬币数目
    }
}
相关推荐
咚为3 小时前
Rust Print 终极指南:从底层原理到全场景实战
开发语言·后端·rust
二哈喇子!3 小时前
基于Spring Boot框架的车库停车管理系统的设计与实现
java·spring boot·后端·计算机毕业设计
Loo国昌6 小时前
【LangChain1.0】第九阶段:文档处理工程 (LlamaIndex)
人工智能·后端·python·算法·langchain
毕设源码-朱学姐6 小时前
【开题答辩全过程】以 基于SpringBoot的律师事务所管理系统的设计与实现为例,包含答辩的问题和答案
java·spring boot·后端
愈努力俞幸运7 小时前
flask 入门 token, headers,cookie
后端·python·flask
毕设源码-朱学姐7 小时前
【开题答辩全过程】以 基于springboot的日用药品仓库管理系统的设计与实现为例,包含答辩的问题和答案
java·spring boot·后端
lkbhua莱克瓦248 小时前
深入理解HTTP协议:从理论到SpringBoot实践
网络·笔记·后端·网络协议·http·javaweb
古城小栈9 小时前
Rust复合类型 四大军阀:数、元、切、串
开发语言·后端·rust
+VX:Fegn089510 小时前
计算机毕业设计|基于springboot + vue酒店预订系统(源码+数据库+文档)
数据库·vue.js·spring boot·后端·课程设计
JavaGuide11 小时前
IntelliJ IDEA 2026.1 EAP 发布!拥抱 Java 26,Spring Boot 4 深度支持!
java·后端·mysql·springboot·idea·大厂面试·javaguide