zoukankan      html  css  js  c++  java
  • LC 431. Encode N-ary Tree to Binary Tree 【lock,hard】

    Design an algorithm to encode an N-ary tree into a binary tree and decode the binary tree to get the original N-ary tree. An N-ary tree is a rooted tree in which each node has no more than N children. Similarly, a binary tree is a rooted tree in which each node has no more than 2 children. There is no restriction on how your encode/decode algorithm should work. You just need to ensure that an N-ary tree can be encoded to a binary tree and this binary tree can be decoded to the original N-nary tree structure.

    For example, you may encode the following 3-ary tree to a binary tree in this way:

     

     

    Note that the above is just an example which might or might not work. You do not necessarily need to follow this format, so please be creative and come up with different approaches yourself.

     

    Note:

    1. N is in the range of [1, 1000]
    2. Do not use class member/global/static variables to store states. Your encode and decode algorithms should be stateless.
    class Codec {
    public:
    
        TreeNode * encode(Node* root) {
            if (!root) return nullptr;
            TreeNode* ret = new TreeNode(root->val);
            TreeNode* tmp = ret;
            if (root->children.size() != 0) {
                tmp->left = encode(root->children[0]);
            }
            tmp = tmp->left;
            for (int i = 1; i < root->children.size(); i++) {
                tmp->right = encode(root->children[i]);
                tmp = tmp->right;
            }
            return ret;
        }
        Node* decode(TreeNode* root) {
            if (!root) return nullptr;
            Node* ret = new Node(root->val, vector<Node*>());
            TreeNode*tmp = root->left;
            while (tmp) {
                ret->children.push_back(decode(tmp));
                tmp = tmp->right;
            }
            return ret;
        }
    };
  • 相关阅读:
    python模块整理9ini配置ConfigParse模块
    python模块整理12pdb调试模块
    django临时
    django实战1使用视图和模板显示多行
    python模块整理10xml.dom.minidom模块
    django_book学习笔记1django介绍
    构建之法阅读笔记 01
    人月神话阅读笔记 06
    人月神话阅读笔记 05
    第四周学习进度
  • 原文地址:https://www.cnblogs.com/ethanhong/p/10158570.html
Copyright © 2011-2022 走看看