zoukankan      html  css  js  c++  java
  • 得到二叉树的每个树枝

    我一只觉得是不是一个深度优先搜索啥的;然后一看是二叉树,立刻脑子就往递归去了,回溯?让叶子往根回溯?通过推迟求值来得到叶子再回溯,这就是我一开始的想法。

    然而却毫无进展,唉,于是我就用了一个很笨的方法,就是每遇到一个右孩子,就复制分叉前的树枝存入容器里,存入的位置由 index 作为其 id为了不会存入错误的位置,新开辟的分支由 size() 做标志。 一开始还是有一个 bug,就是在没有左孩子的情况下,会多出一个没有左孩子的版本。于是加了一个判断,若没有左孩子,则将该分支设为空,并在最后的遍历中删除。在这之前还有一个 bug,就是在遍历完最左边的的分支之后,接下来的分支都包括之前遍历过的左边的分支,后来发现这是求值顺序的问题,只要先求右子树就可以解决。代码如下:

    vector<string> binaryTreePaths(TreeNode* root)
    {
        if (root == nullptr){
            return vector<string>();
        }
        
        bool isRootInited = false;
        vector<string> paths;
        function<void(TreeNode*, size_t)> traversal;
        traversal = [&](TreeNode* node, size_t index)
        {
            if (isRootInited){
                paths[index].append("->" + to_string(node->val));
            }
            else{
                paths.push_back(to_string(node->val));
                isRootInited = true;
            }
            
            if (node->left == nullptr && node->right == nullptr){
                return;
            }
            
            if (node->right != nullptr){
                auto newBranch = paths[index];
                paths.push_back(newBranch);
                traversal(node->right, paths.size() - 1);
            }
            
            if (node->left != nullptr){
                traversal(node->left, index);
            }
            else{
                paths[index] = "";
            }
            
        };
        traversal(root, 0);
        for (auto it = paths.begin(); it != paths.cend(); it->empty()? it = paths.erase(it) : ++it){}
        return paths;
    }
  • 相关阅读:
    浅析Go中的MPG模式(一)
    panic: assignment to entry in nil map
    Golang 新手可能会踩的 50 个坑
    小刘的go面试题
    go 单元测试整理
    go test 测试单个文件和测试单个函数
    mac pro锁屏后没有声音了怎么处理
    go json返回时间字符串处理time.Time类型
    php求一个字符串中不重复的最长子串
    业务订单号生成算法,每秒50W左右,不同机器保证不重复,包含日期可读性好
  • 原文地址:https://www.cnblogs.com/wuOverflow/p/4776196.html
Copyright © 2011-2022 走看看