编程题—二叉树

内容会持续更新，有错误的地方欢迎指正，谢谢!

题目

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

分析

先遍历二叉树，记录每一个叶结点的权值val和编码 。再求出根节点到最大权值结点和最小权值结点的路径（找路径），并用两个变量s1和s2记录两个节点的位置，并求出s1和s2的size，如果有重复的路径就减去重复的路径的长度x （去重），最终结果为：size1+size2-2x。时间复杂度就是遍历二叉树的复杂度。

使用的数据结构： map

代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Tree {
public:
    void getCode(map<int, string>&m, TreeNode*root,string s) {
        if (root->left == NULL && root->right == NULL)//判断是否为叶结点
        {
            m[root->val] = s;//记录每一个叶结点的权值val和编码
            return;
        }
        //因为所有的路径都是从根节点出发的，所以往左走记1，往右走记0
        if (root->left) 
            getCode(m, root->left, (s + '1'));
        if (root->right) 
            getCode(m, root->right, (s + '0'));
    }
    int getDis(TreeNode* root) 
    {
        map<int, string> m;//<权值，编码>
        string s;
        getCode(m, root,s);

        //根据map的性质（红黑树性质），即关键字是有序的，
        //所以iter指向最小权值处，end指向最大权值处。
        map<int, string>::iterator iter = m.begin();
        string s1 = iter->second;
        map<int, string>::iterator end = (--m.end());
        string s2 = end->second;

        int i = 0, j = 0;
        for (; i < s1.size() && j < s2.size()&&s1[i] == s2[j]; i++, j++) {}
        return s1.size() - i + s2.size() - j;
    }
};