LeetCode 979 Distribute Coins in Binary Tree

原创于 2019-01-25 21:42:18 发布 · 349 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 3 个专栏收录

9 篇文章

订阅专栏

dfs

2 篇文章

订阅专栏

树

1 篇文章

订阅专栏

本文介绍了一种解决二叉树中节点金币分配问题的算法。通过递归遍历二叉树，统计每个节点的子树节点数与金币总数，计算节点与金币数量的不平衡量，最终求得整个二叉树最小金币移动次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
 /*
 统计以当前节点为根节点，包括以下（其子孙）共有多少个节点，多少个金币，节点-金币个数的差值的绝对值为不平衡量，需要运进去或运出来，
 递归处理，递归的过程中更新答案
 */
class Solution {
public:
	int res = 0;
	pair<int, int> dfs(TreeNode* root) {

		pair<int, int> p;
		p.first = 1;
		p.second = root->val;   
		if (root->left != NULL) {
			pair<int, int> x = dfs(root->left);
			p.first += x.first;
			p.second += x.second;
		}
		if (root->right != NULL) {
			pair<int, int> x = dfs(root->right); 
			p.first += x.first;
			p.second += x.second;
		}
		res += abs(p.first - p.second);
        return p;
	}
	int distributeCoins(TreeNode* root) {
		dfs(root);

		return res;
	}
};