带权值二叉树的路径最长值与路径

最新推荐文章于 2024-11-21 15:20:18 发布

原创最新推荐文章于 2024-11-21 15:20:18 发布 · 2.5k 阅读

4 ·

CC 4.0 BY-SA版权

学习笔记同时被 2 个专栏收录

12 篇文章

订阅专栏

C++

9 篇文章

订阅专栏

这是一篇关于如何在带有权重的二叉树中找到路径最大权重的问题解析。文章提出了两种方法，第一种方法能求出最大权重但无法显示路径，而第二种方法不仅能找出最大权重，还能记录并展示出这条最大权重的路径。

一、Problem

Given a binary tree, every node has a weight, then you need to find out the path that can make the total weight of all nodes it go through is the maximum and show the path. The weight is always positive.

二、Solution

方法一、 (该方法只能找到路径最大值并不能表示最大值的路径）

int getPath(TreeNode* root, int now) {
    if (root != NULL) {
        int lnow = getPath(root->lchild, root->data);
        int rnow = getPath(root->rchild, root->data);
        int nows = lnow > rnow ? lnow : rnow;
        return now + nows;
    } else {
        return now;
    }
}


int findMaxPath(TreeNode *root) {
    if (root != NULL) {
        int lnow = getPath(root->lchild, root->data);
        int rnow = getPath(root->rchild, root->data);
        int now = lnow > rnow ? lnow : rnow;
        return now;
    } else {
        return 0;
    }
}

方法二、（该方法可以找到最大值且可以输出节点数据，不过得设置一个全局变量sum）

int sum = 0;

int height(TreeNode *t) {  //递归求树高
    int hl,hr;
    if(t == NULL) {
        return 0;
    }
    hl = height(t->lchild)+1;               //最下面的左孩子加一
    hr = height(t->rchild)+1;               //最下面的右孩子加一
    return (hl > hr?hl:hr);                 //比较谁大就取谁
}

void longest_path(TreeNode *root) {  //递归求最长路径
    if (t!= NULL) {
        cout << root->data << " ";  //  输出节点数据
        sum += root->data;// 计算路径最大值                
        if (height(root->lchild) > height(root->rchild)) {   
            longest_path(root->lchild);
        } else {
            longest_path(root->rchild);
        }
    } 
}

上述两种方法各取所需喽！如有更好的方法欢迎欢迎