二叉树两个节点的最大路径

参考博客：传送门

问题：如何求二叉树两个节点的最大路径

思路：分为两种情况
1）路径经过根节点
2）路径不经过根节点，是其左右子树的最大路径

比较这两种情况，去较大的即为最后结果

#include <iostream>

using namespace std;

struct Node{
    Node *Left;
    Node *Right;
};
struct Result{
    int maxdepth;
    int maxDistance;
};
Result GetMaxDistance(Node *root){
    if(!root){
        Result ans={0,0};
        return ans;
    }
    Result lans = GetMaxDistance(root->Left);
    Result rans = GetMaxDistance(root->Right);
    Result result;
    result.maxdepth = max(lans.maxdepth,rans.maxdepth)+1;
    result.maxDistance = max(max(lans.maxDistance,rans.maxDistance),lans.maxdepth+rans.maxdepth);
    return result;
}
void Link(Node *nodes,int p,int l,int r){
    if(l!=-1){
        nodes[p].Left = &nodes[l];
    }
    if(r!=-1){
        nodes[p].Right = &nodes[r];
    }
}
int main()
{
    Node test1[9] = { 0 };
    Link(test1, 0, 1, 2);
    Link(test1, 1, 3, 4);
    Link(test1, 2, 5, 6);
    Link(test1, 3, 7, -1);
    Link(test1, 5, -1, 8);
    cout<<"test1:"<<GetMaxDistance(&test1[0]).maxDistance<<endl;

    Node test2[4] = { 0 };
    Link(test2, 0, 1, 2);
    Link(test2, 1, 3, -1);
    cout<<"test2:"<<GetMaxDistance(&test2[0]).maxDistance<<endl;

    Node test3[9] = { 0 };
    Link(test3, 0, -1, 1);
    Link(test3, 1, 2, 3);
    Link(test3, 2, 4, -1);
    Link(test3, 3, 5, 6);
    Link(test3, 4, 7, -1);
    Link(test3, 5, -1, 8);
    cout<<"test3:"<<GetMaxDistance(&test3[0]).maxDistance<<endl;

    Node test4[9] = { 0 };
    Link(test4, 0, 1, 2);
    Link(test4, 1, 3, 4);
    Link(test4, 3, 5, 6);
    Link(test4, 5, 7, -1);
    Link(test4, 6, -1, 8);
    cout<<"test4:"<<GetMaxDistance(&test4[0]).maxDistance<<endl;
    return 0;
}