两个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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
如果 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) {}
* };
*/