leetcode算法题--根据描述创建二叉树

最新推荐文章于 2022-08-05 16:27:43 发布

bob62856

最新推荐文章于 2022-08-05 16:27:43 发布

阅读量296

点赞数

分类专栏： Algorithm 文章标签： leetcode 算法深度优先

本文链接：https://blog.csdn.net/qq_20817327/article/details/124785921

版权

Algorithm 专栏收录该内容

374 篇文章 3 订阅

订阅专栏

原题链接：https://leetcode.cn/problems/create-binary-tree-from-descriptions/

class Solution {
public:
    TreeNode* createBinaryTree(vector<vector<int>>& descriptions) {
        unordered_map<int, TreeNode*> mp;
        unordered_map<int, bool> isRoot;
        for (auto desc : descriptions) {
            int p = desc[0], c = desc[1];
            if (!mp.count(p)) {
                mp[p] = new TreeNode(p);
            }
            if (!mp.count(c)) {
                mp[c] = new TreeNode(c);
            }
            if (!isRoot.count(p)) {
                isRoot[p] = true;
            }
            isRoot[c] = false;
            TreeNode* node = mp[p];
            TreeNode* child = mp[c];
            if (desc[2]) {
                node->left = child;
            } else {
                node->right = child;
            }
        }
        for (auto [k, v] : isRoot) {
            if (v) return mp[k];
        }
        return NULL;  
    }
};