2023-07-14力扣每日一题

链接：

979. 在二叉树中分配硬币

题意：

一个二叉树，n个节点，节点权值总和为n，每次可以相邻节点间移动1权值

求让每个节点都为1的最少次数

解：

给定了一个树的结构体，先整一手DFS/BFS，n不大，随便莽莽

首先每个节点只需要剩下1，而且可以知道叶子节点如果不为1，肯定要通过父节点移动

所以只要预设该叶子结点为1，所需要的代价转移到父节点身上，就可以将该叶子结点从树中抹去（因为已经处理完毕）

分析以下几种情况：

1.该叶子结点值为1：由于是叶子结点，且数量正好，所以发生移动的可能性为0，代表这个叶子节点已经处理完毕，可以从树中抹去不在考虑

2.该叶子结点值小于1：由于具有唯一的来源（父节点）所以可以进行代价转移，先花费1-val次移动次数，同时使父节点的值减少1-val，而后将自身抹去，因为不再参与移动（由于父节点可能为0，减少后存在负数，方便理解，使用权值表示而不是金币）

3.该叶子结点值大于1：由于具有唯一的去向（父节点）所以可以进行代价转移，先花费val-1次移动次数，将多余权值全部给予父节点，而后将自身抹去，因为不再参与移动

4.当所有叶子结点从树中抹去，则原先它们的父节点称为新的叶子结点，在进行前面三种情况的处理。

5.最后只剩根节点的时候，由于权值总和为n，当每个节点为1的时候才抹去，则处理到根节点时值一定为1

实际代码：

#include
using namespace std;
struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
void add(vector &temp,TreeNode *root,int mao)//建树 
{
    root->val=temp[mao-1];
    if( mao*2-1left=new TreeNode;
        add(temp,root->left,mao*2);
    }
    
    if( mao*2right=new TreeNode;
        add(temp,root->right,mao*2+1);
    }
}
void solve(TreeNode* now,int &ans,TreeNode* father)
{
    if(now->left!=nullptr) solve(now->left,ans,now);//DFS
    
    if(now->right!=nullptr) solve(now->right,ans,now);//DFS
    
    //if(father==nullptr) return;
    //最后只剩根节点的时候，由于权值总和为n，当每个节点为1的时候才抹去，则处理到根节点时值一定为1
    //cout<val<<" "<val<1 && father!=nullptr)
    {
        ans+= 1-now->val;//计算移动次数 
        father->val-= 1-now->val;//代价转移 
        now->val=1;//设为1
    }
    
    if(now->val>1 && father!=nullptr)
    {
        ans+= now->val-1;//计算移动次数 
        father->val+= now->val-1;//代价转移
        now->val=1;//设为1
    }
}
int distributeCoins(TreeNode* root)
{
    int ans=0;
    solve(root,ans,nullptr);
    return ans;
}
int main()
{
    int n;cin>>n;
    vector temp;TreeNode root;
    for(int i=1;i<=n;i++)
    {
        int t;cin>>t;temp.push_back(t);
    }
    add(temp,&root,1);//建树 
    
    int ans=distributeCoins(&root);
    cout<

限制：

• 树中节点的数目为 n
• 1 <= n <= 100
• 0 <= Node.val <= n
• 所有 Node.val 的值之和是 n