465二叉树分割最大差值(递归)

1、题目描述

给出一颗二叉树,每个节点有一个编号和个值,该值可能为负数,请你找出个最优节点(除根节点外),使得在该节点将树分成两棵树后(原来的树移除这个节点及其子节点,新的树以该节点为根节点),分成的两棵树各节点的和之间的差绝对值最大。请输出该节点编号,如有多个相同的差,输出编号最小的节点。

2、示例

输入:

4
4 9 -7 -8
0 1
0 3
1 2
第一行,四个节点,编号0-3,范围1-10000
第二行,节点0-3权值
第三行到第五行,表示二叉树各节点间的父子关系
0 1 节点0的左节点是1
0 3 节点0的右节点是3
1 2 节点1的左节点是2
注意:左节点永远出现在右节点之前

输出:

节点编号,示例中编号为3的节点是最优节点

3、题解

基本思想:递归,先递归求出每个节点的子树节点之和,然后遍历每个节点,求出diffvalue=abs(node.second.sum*2-sum)是否大于maxdiffvalue,如果是更新maxdiffvalue以及targetnode。

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<math.h>
using namespace std;
struct node{
    int num_child;
    int value;
    int sum;
    int child[2];
};
int sum=0;
map<int,node> node_map;
int caculate_sum(int root)
{
    int ret=node_map[root].value;
    for(int i=0;i<node_map[root].num_child;i++)
    {
        int x=caculate_sum(node_map[root].child[i]);
        ret+=x;
    }
    node_map[root].sum=ret;
    return ret;
}
int main()
{
    int n,root;
    vector<bool> isroot;
    cin>>n;
    isroot=vector<bool>(n,true);
    node tmp;
    for(int i=0;i<n;i++)
    {
        int x;
        cin>>x;
        node_map[i].num_child=0;
        node_map[i].value=x;
        sum+=x;
    }
    for(int i=0;i<n-1;i++)
    {
        int fa,child;
        cin>>fa>>child;
        isroot[child]=false;
        node_map[fa].child[node_map[fa].num_child++]=child;
    }
    for(int i=0;i<n;i++)
    {
        if(isroot[i])
        {
            root=i;
            break;
        }
    }
    caculate_sum(root);
    int maxm=0;
    int index=0;
    for(auto &i:node_map)
    {
        int cha=2*i.second.sum-sum;
        if(maxm<abs(cha)){
            maxm=abs(cha);
            index=i.first;
        }
        else if(maxm==abs(cha)&&i.first<index)
        {
            index=i.first;
        }
    }
    cout<<index<<endl;
    return 0;
}

 

  • 6
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值