代码随想录算法训练营第18天|517.找树左下角的值、路径总和、从中序和后序遍历序列构造二叉树

代码随想录算法训练营第18天|517.找树左下角的值、路径总和、从中序和后序遍历序列构造二叉树

517.找树左下角的值

树左下角的值，不一定是直接往左子树遍历就行。

// 递归
class Solution {
public:
    int maxDepth = INT_MIN;
    int res;
    void traversal(TreeNode* root, int depth) {
        if (root->left == nullptr && root->right == nullptr) {
            if (depth > maxDepth) {
                maxDepth = depth;
                res = root->val;
            }
            return ;
        }
        if (root->left == nullptr) {
            ++depth;
            traversal(root->left, depth);
            --depth;
        }
        if (root->right == nullptr) {
            ++depth;
            traversal(root->left, depth);
            --depth;
        }
        return ;
    }
    int findBottomLeftValue(TreeNode* root) {
        traversal(root, 0);
        return res;
    }
};

// 迭代法
class Solution{
public:
   int findBottomLeftValue(TreeNode* root) {
       queue<TreeNode*> que;
       if (root != nullptr) que.push(root);
       int res = 0;
       while (!que.empty()) {
           int size = que.size();
           for (int i = 0; i < size; ++i) {
               TreeNode* node = que.front();
               que.pop();
               if(i == 0) res = node->val;  //记录每行第一个元素
               if(node->left != nullptr) que.push(node->left);
               if(node->right != nullptr) que.push(node->right);
           }
       }
       return res;
   }
};
 

112.路径总和

不需要遍历整个树的所有节点，只要找到了符合要求的，直接返回即可。

使用目标值减去节点值的做法，最后减到0，则说明是true

// 递归
class Solution {
private:
    bool traversal(TreeNode* cur, int count) {
        if(cur->left == nullptr && cur->right == nullptr && count == 0) return true;  //遇到叶子节点，且数字为0
        if(cur->left == nullptr && cur->right == nullptr) return false; //叶子节点，数字不为0
        if(cur->left != nullptr) {  //左
            count -= cur->left->val;  //递归，处理节点
            if(traversal(cur->left, count)) return true;
            count += cur->left->val;  //回溯
        }
        if(cur->right) {
            count -= cur->right->val;
            if(traversal(cur->right, count)) return true;
            count += cur->right->val;  //回溯
        }
        return false;
    }
public:
    bool hasPathSum(TreeNode* root, int sum) {
        if(root == NULL) return false;
        return traversal(root, sum - root->val);
    }
};

113.路径总和Ⅱ

因为需要找出所有的路径，有别于上一题，所以就需要遍历所有的节点. 递归函数不需要返回值

// 递归
class Solution {
private:
    vector<vector<int>> res;
    vector<int> path;
    void traversal(TreeNode* cur, int count) {
        if (cur->left == nullptr && cur->right == nullptr && count == 0) { // 遇到了叶子节点且找到了和为sum的路径
            res.push_back(path);
            return;
        }

        if (cur->left == nullptr && cur->right == nullptr) return ; // 遇到叶子节点而没有找到合适的边，直接返回

        if (cur->left != nullptr) { // 左 （空节点不遍历）
            path.push_back(cur->left->val);
            count -= cur->left->val;
            traversal(cur->left, count);    // 递归
            count += cur->left->val;        // 回溯
            path.pop_back();                // 回溯
        }
        if (cur->right != nullptr) { // 右 （空节点不遍历）
            path.push_back(cur->right->val);
            count -= cur->right->val;
            traversal(cur->right, count);   // 递归
            count += cur->right->val;       // 回溯
            path.pop_back();                // 回溯
        }
        return ;
    }

public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        res.clear();
        path.clear();
        if (root == nullptr) return res;
        path.push_back(root->val); // 把根节点放进路径
        traversal(root, sum - root->val);
        return res;
    }
};

106.从中序和后序遍历序列构造二叉树

class Solution {
private:
    TreeNode* traversal (vector<int>& inorder, vector<int>& postorder) {

        // 后序遍历数组最后一个元素，就是当前的中间节点
        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 nullptr;
        return traversal(inorder, postorder);
    }
};

105.从前序和中序遍历序列构造二叉树

class Solution {
private:
        TreeNode* traversal (vector<int>& inorder, int inorderBegin, int inorderEnd, vector<int>& preorder, int preorderBegin, int preorderEnd) {
        if (preorderBegin == preorderEnd) return nullptr;

         // 注意用preorderBegin 不要用0
        int rootValue = preorder[preorderBegin];
        TreeNode* root = new TreeNode(rootValue);

        if (preorderEnd - preorderBegin == 1) return root;

        int delimiterIndex;
        for (delimiterIndex = inorderBegin; delimiterIndex < inorderEnd; delimiterIndex++) {
            if (inorder[delimiterIndex] == rootValue) break;
        }
        // 切割中序数组
        // 中序左区间，左闭右开[leftInorderBegin, leftInorderEnd)
        int leftInorderBegin = inorderBegin;
        int leftInorderEnd = delimiterIndex;
        // 中序右区间，左闭右开[rightInorderBegin,rightInorderEnd)
        int rightInorderBegin = delimiterIndex + 1;
        int rightInorderEnd = inorderEnd;

        // 切割前序数组
        // 前序左区间，左闭右开[leftPreorderBegin,leftPreorderEnd)
        int leftPreorderBegin =  preorderBegin + 1;
        // 终止位置是起始位置加上中序左区间的大小size
        int leftPreorderEnd = preorderBegin + 1 + delimiterIndex - inorderBegin;
        // 前序右区间, 左闭右开 [rightPreorderBegin,rightPreorderEnd)
        int rightPreorderBegin = preorderBegin + 1 + (delimiterIndex - inorderBegin);
        int rightPreorderEnd = preorderEnd;

        root->left = traversal(inorder, leftInorderBegin, leftInorderEnd,  preorder, leftPreorderBegin, leftPreorderEnd);
        root->right = traversal(inorder, rightInorderBegin, rightInorderEnd, preorder, rightPreorderBegin, rightPreorderEnd);
        return root;
    }

public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if (inorder.size() == 0 || preorder.size() == 0) return nullptr;

        // 参数坚持左闭右开的原则
        return traversal(inorder, 0, inorder.size(), preorder, 0, preorder.size());
    }
};