码迷,mamicode.com
首页 > 编程语言 > 详细

(算法)二叉树两个结点的最远距离

时间:2015-08-02 16:44:56      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:

题目:

求二叉树两个结点的最远距离。

二叉树定义如下:

class TreeNode{
public:
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x):val(x),left(NULL),right(NULL){}
};

思路:

遍历每个节点,找出以当前节点为根的最长路径,然后找出所有最长路径中的最大值。

代码:

代码1:

class Node{
public:
    int val;
    Node* left;
    Node* right;
    int maxLeft;
    int maxRight;
};

void longestPath_1(Node* pRoot,int& maxLen){
    if(pRoot==NULL)
        return;
    if(pRoot->left==NULL)
        pRoot->maxLeft=0;
    if(pRoot->right==NULL)
        pRoot->maxRight=0;

    int leftLen;
    if(pRoot->left){
        longestPath_1(pRoot->left,maxLen);
        leftLen=1+max(pRoot->left->maxLeft,pRoot->left->maxRight);
        pRoot->maxLeft=leftLen;
    }

    int rightLen;
    if(pRoot->right){
        longestPath_1(pRoot->right,maxLen);
        rightLen=1+max(pRoot->right->maxLeft,pRoot->right->maxRight);
        pRoot->maxRight=rightLen;
    }

    maxLen=max(pRoot->maxLeft+pRoot->maxRight,maxLen);
}

代码2:

class TreeNode{
public:
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x):val(x),left(NULL),right(NULL){}
};

void longestPath(TreeNode* pRoot,int& maxLeft,int& maxRight,int& maxLen){
    if(pRoot==NULL){
        maxLeft=0;
        maxRight=0;
        maxLen=0;
        return;
    }

    int maxLeft_1,maxLeft_2,maxRight_1,maxRight_2;
    longestPath(pRoot->left,maxLeft_1,maxRight_1,maxLen);
    longestPath(pRoot->right,maxLeft_2,maxRight_2,maxLen);

    maxLeft=1+max(maxLeft_1,maxRight_1);
    maxRight=1+max(maxLeft_2,maxRight_2);

    maxLen=max(maxLeft,maxRight)+1;
}

(算法)二叉树两个结点的最远距离

标签:

原文地址:http://www.cnblogs.com/AndyJee/p/4695813.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!