网易2016 实习研发工程师 [编程题]二叉树

传送门

有一棵二叉树，树上每个点标有权值，权值各不相同，请设计一个算法算出权值最大的叶节点到权值最小的叶节点的距离。二叉树每条边的距离为1，一个节点经过多少条边到达另一个节点为这两个节点之间的距离。

给定二叉树的根节点root，请返回所求距离。

题解：

给每个节点编号（0 - total），用map记录每个节点的父节点

找出最大权值叶子节点 和 最小权值 叶子节点的编号

然后就是寻找最近公共祖先了

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class Tree {
public:
    int getDis(TreeNode* root) {
        // write code here
        if(root == NULL) return 0;
        int ma = INT_MIN,mi = INT_MAX;
        int mapos = -1, mipos = -1;
        int total = 0;
        map<int,int> parent;
        parent[0] = -1;
        dfs(root,0,total,parent,ma,mi,mapos,mipos);
        int ans = 0;
        while(mapos != mipos) {
            if(mapos > mipos){
                mapos = parent[mapos];
            }
            else{
                mipos = parent[mipos];
            }
            ans ++;
        }
        return ans;
    }
    
    void dfs(TreeNode* root,int index,int &total,map<int,int> &parent,int &ma,int &mi,int &mapos,int &mipos) {
        if(root == NULL) {
            return;
        }
        if(root->left == NULL && root->right == NULL){
            if(root->val < mi) {
                mi = root->val;
                mipos = index;
            }
            if(root->val > ma) {
                ma = root->val;
                mapos = index;
            }
            return;
        }
        if(root->left != NULL) {
            total++;
            parent[total] = index;
            dfs(root->left,total,total,parent,ma,mi,mapos,mipos);
        }
        if(root->right != NULL) {
            total++;
            parent[total] = index;
            dfs(root->right,total,total,parent,ma,mi,mapos,mipos);
        }
    }
    
};