两个unordered_map,一个用于构造树,另一个用于记录子结点,寻找根结点

注意细节
class Solution {
public:
    void myinput(unordered_map<int,TreeNode*> &m,int f,int c,int b){
        if(m.count(f)==1) {
            if(m.count(c)==1) {
                if(b==1) m[f]->left=m[c];
                else m[f]->right=m[c];
            }
            else{
                TreeNode*p =new TreeNode(c);//用于创建一个val为c的结点,将结点的地址返回给p;
                m.insert(make_pair(c,p));//
                if(b==1) m[f]->left=m[c];
                else m[f]->right=m[c];
            }
        }
            else {
                TreeNode*p1 =new TreeNode(f);
                m.insert(make_pair(f,p1));
            if(m.count(c)==1) {
                if(b==1) m[f]->left=m[c];
                else m[f]->right=m[c];
            }
            else{
                TreeNode*p =new TreeNode(c);
                m.insert(make_pair(c,p));//
                if(b==1) m[f]->left=m[c];
                else m[f]->right=m[c];
            }
            }
    }
    TreeNode* createBinaryTree(vector<vector<int>>& de) {
        unordered_map<int,TreeNode*> m;
        unordered_map<int,bool> m2;
        for(int i=0;i<de.size();i++){
            myinput(m,de[i][0],de[i][1],de[i][2]);
            if(m2.count(de[i][0])==0) {m2.insert(make_pair(de[i][0],0));//cout<<de[i][0]<<" ";
            }
            if(m2.count(de[i][1])==0) m2.insert(make_pair(de[i][1],1));
            else m2[de[i][1]]=1;
        }
       // cout<<endl;
        for(auto i:m2) {
            //cout<<i.first<<" "<<i.second<<" "<<m[i.first]->val<<" ";
            if(i.second==0) return m[i.first];
            }
        return nullptr;
    }
}
给你一个二维整数数组 descriptions ,其中 descriptions[i] = [parenti, childi, isLefti] 表示 parenti 是 childi 在 二叉树 中的 父节点,二叉树中各节点的值 互不相同 。此外:
如果 isLefti == 1 ,那么 childi 就是 parenti 的左子节点。
如果 isLefti == 0 ,那么 childi 就是 parenti 的右子节点。
请你根据 descriptions 的描述来构造二叉树并返回其 根节点 。
测试用例会保证可以构造出 有效 的二叉树。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/create-binary-tree-from-descriptions
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务