两个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) {}
* };
*/
查看29道真题和解析