算法训练第十八天|513.找树左下角的值、112. 路径总和、106.构造二叉树

二叉树|513.找树左下角的值、112. 路径总和、106.构造二叉树


513.找树左下角的值

自己审题思路

层序遍历找最后一层第一个元素

看完代码随想录题解后的收获

递归法的学习

代码:
class Solution {
public:
    int findBottomLeftValue(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        int result = 0;
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                if (i == 0) result = node->val; // 记录每一行第一个元素(循环覆盖,最终记录最后一行第一个元素)
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return result;
    }
};

代码(递归)

class Solution {
public:
    int maxDepth = INT_MIN;
    int result;

    void traversal(TreeNode* cur, int depth) {
        if(!cur->left && !cur->right) {
            if(depth > maxDepth) {  
                maxDepth = depth;   // 更新最大深度
                result = cur->val;  // 最大深度最左面的数值
            }
            return;
        }

        if(cur->left) traversal(cur->left, depth + 1);
        if(cur->right) traversal(cur->right, depth + 1);
        return;
    }

    int findBottomLeftValue(TreeNode* root) {
        if(root == nullptr) return 0;
        traversal(root, 0);
        return result;
    }
};

参考详解


112. 路径总和

自己审题思路

看完代码随想录题解后的收获

递归函数什么时候要有返回值,什么时候没有返回值(三点)

  • 如果需要搜索整棵二叉树且不用处理递归返回值,递归函数就不要返回值。(113.路径总和ii)
  • 如果需要搜索整棵二叉树且需要处理递归返回值,递归函数就需要返回值。 (236. 二叉树的最近公共祖先
  • 如果要搜索其中一条符合条件的路径,那么递归一定需要返回值,因为遇到符合条件的路径了就要及时返回。(本题的情况)
代码:
class Solution {
public:
    bool traversal(TreeNode* cur, int count) {
        if(!cur->left && !cur->right && count == 0) return true;
        if(!cur->left && !cur->right) return false;

        if (cur->left) {
            if(traversal(cur->left,count - cur->left->val)) return true;
        }
        if (cur->right) {
            if(traversal(cur->right,count - cur->right->val)) return true;
        }
        return false;
    }

    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root == nullptr) return false;
        return traversal(root,  targetSum - root->val);
    }
};
/**
 * 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:
    bool traversal(TreeNode* cur, int count) {
        if(!cur->left && !cur->right && count == cur->val) return true;
        if(!cur->left && !cur->right) return false;

        if (cur->left) {
            if(traversal(cur->left,count - cur->val)) return true;
        }
        if (cur->right) {
            if(traversal(cur->right,count - cur->val)) return true;
        }
        return false;
    }

    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root == nullptr) return false;
        return traversal(root,  targetSum);
    }
};

参考详解


构造二叉树

自己审题思路

这道题有思路:根据中序找到根节点,然后循环找,但是写代码的时候遇到不少问题。

看完代码随想录题解后的收获

理解思路和debug。

代码:
class Solution {
private:
    TreeNode* traversal (vector<int>& inorder, vector<int>& postorder) {
        if (postorder.size() == 0) return NULL;

        // 后序遍历数组最后一个元素,就是当前的中间节点
        int rootValue = postorder[postorder.size() - 1];
        TreeNode* root = new TreeNode(rootValue);

        // 叶子节点
        if (postorder.size() == 1) return root;

        // 找到中序遍历的切割点
        int delimiterIndex;
        for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++) {
            if (inorder[delimiterIndex] == rootValue) break;
        }

        // 切割中序数组
        // 左闭右开区间:[0, delimiterIndex)
        vector<int> leftInorder(inorder.begin(), inorder.begin() + delimiterIndex);
        // [delimiterIndex + 1, end)
        vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end() );

        // postorder 舍弃末尾元素
        postorder.resize(postorder.size() - 1);

        // 切割后序数组
        // 依然左闭右开,注意这里使用了左中序数组大小作为切割点
        // [0, leftInorder.size)
        vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
        // [leftInorder.size(), end)
        vector<int> rightPostorder(postorder.begin() + leftInorder.size(), postorder.end());

        root->left = traversal(leftInorder, leftPostorder);
        root->right = traversal(rightInorder, rightPostorder);

        return root;
    }
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if (inorder.size() == 0 || postorder.size() == 0) return NULL;
        return traversal(inorder, postorder);
    }
};
class Solution {
public:
    TreeNode* dfs(vector<int>& inorder, int inStart, int inEnd, vector<int>& postorder, int postStart, int postEnd) {
        if(inStart == inEnd ) return nullptr;

        int rootValue = postorder[postEnd - 1];
        int split;
        for (int i = 0; i < inorder.size(); i++) {
            if (rootValue == inorder[i]) {
                split = i;
                break;
            }
        }
        TreeNode* root = new TreeNode(rootValue);
        int inLeftStart = inStart;
        int inLeftEnd = split;

        int inRightStart = split + 1;
        int inRightEnd = inEnd;

        int postLeftStart = postStart;
        int postLeftEnd = postLeftStart + (inLeftEnd - inLeftStart);

        int postRightStart = postLeftEnd;
        int postRightEnd = postEnd - 1;

        root->left = dfs(inorder, inLeftStart, inLeftEnd, postorder, postLeftStart, postLeftEnd);
        root->right = dfs(inorder, inRightStart, inRightEnd, postorder, postRightStart, postRightEnd);

        return root;
    }

    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if(inorder.size() == 0 || postorder.size() == 0) return nullptr;
        return dfs(inorder, 0, inorder.size(), postorder, 0, postorder.size());
    }
};

参考详解


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容