LeetCode 2196. 根据描述创建二叉树（哈希）

Michael阿明

发布于 2022-03-10 18:31:23

3120

发布于 2022-03-10 18:31:23

文章目录
1. 题目
2. 解题

1. 题目

给你一个二维整数数组 descriptions ，其中 descriptions[i] = [parenti, childi, isLefti] 表示 parenti 是 childi 在二叉树中的 父节点，二叉树中各节点的值互不相同。此外：

如果 isLefti == 1 ，那么 childi 就是 parenti 的左子节点。
如果 isLefti == 0 ，那么 childi 就是 parenti 的右子节点。

请你根据 descriptions 的描述来构造二叉树并返回其 根节点 。

测试用例会保证可以构造出有效的二叉树。

示例 1：

输入：descriptions = [[20,15,1],[20,17,0],[50,20,1],[50,80,0],[80,19,1]]
输出：[50,20,80,15,17,19]
解释：根节点是值为 50 的节点，因为它没有父节点。
结果二叉树如上图所示。

示例 2：

输入：descriptions = [[1,2,1],[2,3,0],[3,4,1]]
输出：[1,2,null,null,3,4]
解释：根节点是值为 1 的节点，因为它没有父节点。 
结果二叉树如上图所示。 
 
提示：
1 <= descriptions.length <= 10^4
descriptions[i].length == 3
1 <= parenti, childi <= 10^5
0 <= isLefti <= 1
descriptions 所描述的二叉树是一棵有效二叉树

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/create-binary-tree-from-descriptions 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

哈希表存储节点的地址，节点的入度，入度为0的是根节点

/**
 * 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) {}
 * };
 */
class Solution {
public:
    TreeNode* createBinaryTree(vector<vector<int>>& descriptions) {
        unordered_map<int, TreeNode*> v2p;
        unordered_map<int, int> indegree;
        for(auto& d : descriptions)
        {
            TreeNode *p = NULL, *c = NULL;
            if(v2p.find(d[0]) != v2p.end())
                p = v2p[d[0]];
            else
            {
                p = new TreeNode(d[0]);
                v2p[d[0]] = p;
            }
            if(v2p.find(d[1]) != v2p.end())
                c = v2p[d[1]];
            else
            {
                c = new TreeNode(d[1]);
                v2p[d[1]] = c;
            }
            if(d[2])
                p->left = c;
            else
                p->right = c;
            if(indegree.find(d[0]) == indegree.end())
                indegree[d[0]] = 0;
            indegree[d[1]]++;
        }
        for(auto& item : indegree)
        {
            if(item.second == 0)
                return v2p[item.first];
        }
        return NULL;
    }
};

832 ms 271.2 MB C++

我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

二叉树