1161. 最大层内元素和

1161. 最大层内元素和

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层,而根节点的子节点位于第 2 层,依此类推。

请你找出层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。

示例:

输入:[1,7,0,7,-8,null,null]
输出:2
解释:
第 1 层各元素之和为 1,
第 2 层各元素之和为 7 + 0 = 7,
第 3 层各元素之和为 7 + -8 = -1,
所以我们返回第 2 层的层号,它的层内元素之和最大。

代码:

int maxvalue=-2147648;
int findMaxStep(BTree bt){

    queue<Node* >que;
    Node* front;
    int sum=bt->val;
    int curstep=0;
    int mystep;
    que.push(bt);
    while(!que.empty()){
        front= que.front();
        que.pop();
        if(curstep<front->step){
            if(maxvalue<sum){
                maxvalue = sum;
                mystep=curstep;
            }
            curstep=front->step;
            sum=0;
        }
        if(front->left) que.push(front->left);
        if(front->right)que.push(front->right);
        sum+=front->val;
    }
    if(sum>maxvalue){
        mystep=front->step;
    }
    cout<<mystep<<endl;
}

 

posted @ 2020-09-26 15:18  图神经网络  阅读(98)  评论(0编辑  收藏  举报
Live2D