DAY18:二叉树提升513、112、113、106、105

文章介绍了如何在LeetCode题目中使用迭代和递归方法解决涉及树结构的问题,如查找树的左下角值、路径总和以及根据中序和后序遍历重建二叉树。
摘要由CSDN通过智能技术生成

Leetcode: 513 找树左下角的值

这道题的思路,其实就是寻找到最后一层的最左边的节点。

迭代法

用迭代法,定义一个变量来保存每一层的第一个节点的数值,然后层级遍历下去,直到最后一层。

class Solution {
public:
    int findBottomLeftValue(TreeNode* root) {
        queue<TreeNode*> que;//定义队列
        int result;
        if(root != NULL) que.push(root);//先将根节点输入队列
        while(!que.empty()){
            vector<int> vec;//保存每层的结构
            int nodesize = que.size();//计算每层的大小
            while(nodesize--){
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            result = vec[0];//取每层的第一个结果。最后一层就是我们大答案
        }
        return result;

    }
};

递归法

保证左边优先搜索,按照前序遍历来。找到深度最大的叶子节点就是最后的答案。

  • 递归的输入和输出:输入节点和最大深度
  • 递归结束的条件:遇到叶子节点就统计最大深度
  • 递归的层级逻辑:左边节点判断,右边节点判断,因为不需要对中间节点进行处理,因此直接返回。
class Solution {
public:
    int maxdepth = INT_MIN;
    int result;
    void findvalue(TreeNode* node, int depth){
        if(node->left == NULL && node->right == NULL){
            if(depth > maxdepth){
                maxdepth =depth;
                result = node->val;
            }
            return;
        }
        if(node->left) findvalue(node->left, depth + 1);
        if(node->right) findvalue(node->right, depth + 1);
        return;
    }
    int findBottomLeftValue(TreeNode* root) {
        findvalue(root, 0);
        return result;
    }
};

Leetcode: 112 路径总和

递归法

  • 递归输入输出:输入节点,还需要输入一个count来计算是否达到数值,输出true或者false
  • 递归结束条件:找到一条路径就返回true
  • 单层递归逻辑:遍历
class Solution {
public:
    bool decide(TreeNode* node, int count){
        if(node->left == NULL && node->right == NULL && count == 0) return true;
        if(node->left == NULL && node->right == NULL && count != 0) return false;
        if(node->left){//判断左子树的情况
            count -= node->left->val;
            if(decide(node->left, count)) return true;//如果左子树出现合适的路径,就返回
            count += node->left->val;//回溯

        }
        if(node->right){
            count -= node->right->val;
            if(decide(node->right, count)) return true;
            count += node->right->val;
        }
        return false;

    }
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root == NULL) return false;
        return decide(root, targetSum - root->val);
    }
};

迭代法

这道题迭代法其实更加符合正常的思维。但这题迭代法较为困难,主要是第一次学习pair的使用。判断条件就和上述一样。

class Solution {
public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root == NULL) return false;
        stack<pair<TreeNode*, int>> st;//需要使用一对结构,指针和指向当前节点的总和
        st.push(pair<TreeNode*, int>(root, root->val));
        while(!st.empty()){
            pair<TreeNode*, int> node = st.top();
            st.pop();
            if(node.first->left == NULL && node.first->right == NULL && node.second == targetSum) return true;
            if(node.first->left){
                st.push(pair<TreeNode*, int>(node.first->left, node.second + node.first->left->val));
            }
            if(node.first->right){
                st.push(pair<TreeNode*, int>(node.first->right, node.second + node.first->right->val));
            }
        }
        return false;
    }
};

Leetcode: 113 路径总和ii

下述代码为代码随想录内容,通过这道题可以很好的学习回溯的内容,以及复杂的处理代码随想录

class solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    // 递归函数不需要返回值,因为我们要遍历整个树
    void traversal(TreeNode* cur, int count) {
        if (!cur->left && !cur->right && count == 0) { // 遇到了叶子节点且找到了和为sum的路径
            result.push_back(path);
            return;
        }

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

        if (cur->left) { // 左 (空节点不遍历)
            path.push_back(cur->left->val);
            count -= cur->left->val;
            traversal(cur->left, count);    // 递归
            count += cur->left->val;        // 回溯
            path.pop_back();                // 回溯
        }
        if (cur->right) { // 右 (空节点不遍历)
            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) {
        result.clear();
        path.clear();
        if (root == NULL) return result;
        path.push_back(root->val); // 把根节点放进路径
        traversal(root, sum - root->val);
        return result;
    }
};

Leetcode: 106 从中序与后序遍历序列构造二叉树

中序为左中右、后序为左右中,因此可以先根据后序数组的最后一个元素,判断出根节点,然后再再中序数组中寻找节点,对中序数组进行划分,然后在对后序数组进行划分,继续寻找中间节点。而使用前序与中序遍历,则根据前序数组来寻找中间节点的位置。主要思路可以分为以下:

  1. 如果后序数组大小为0,则是空节点。
  2. 如果后序数组不为空,那么最后一个元素就是中间节点。
  3. 根据后序数组中中间节点的数值,在中序数组中找到中间节点
  4. 对中序数组进行划分
  5. 对后序数组进行划分
  6. 递归处理左右区间
class Solution {
public:
    TreeNode* createtree(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 index = 0;
        for(index = 0; index < inorder.size(); index++){
            if(inorder[index] == rootvalue) break;
        }
        //划分中序数组
        vector<int> leftInorder(inorder.begin(), inorder.begin() + index);
        vector<int> rightInorder(inorder.begin() + index + 1, inorder.end());
    	//划分后序数组
        postorder.resize(postorder.size() - 1);
        vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
        vector<int> rightPostorder(postorder.begin() + leftInorder.size(), postorder.end());
    	//递归
        root->left = createtree(leftInorder, leftPostorder);
        root->right = createtree(rightInorder, rightPostorder);
        return root;

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

    }
};

Leetcode: 105 从前序与中序遍历序列构造二叉树

与上述思路基本相同,但是需要注意划分的时候起始点需要处理。代码为代码随想录

代码随想录

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

        int rootValue = preorder[preorderBegin]; // 注意用preorderBegin 不要用0
        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;
        int leftPreorderEnd = preorderBegin + 1 + delimiterIndex - inorderBegin; // 终止位置是起始位置加上中序左区间的大小size
        // 前序右区间, 左闭右开[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 NULL;

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

前序和后序无法确定一棵树。

  • 6
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值