LeetCode 2196. 根据描述创建二叉树(哈希)

2022-03-10 18:31:23 浏览数 (1)

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

1. 题目

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

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

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

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

示例 1:

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

示例 2:

代码语言:javascript复制
输入: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的是根节点
代码语言:javascript复制
/**
 * 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/

0 人点赞