C++牛客网编程(十)

目录

判断数组是否是某二叉树的后序遍历

二叉树中和为某一值的路径(一)

二叉树中和为某一值得路径(二)


判断数组是否是某二叉树的后序遍历

描述

输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则返回 true ,否则返回 false 。假设输入的数组的任意两个数字都互不相同。

数据范围: 节点数量 0≤n≤10000节点上的值满足 1≤val≤1051 ,保证节点上的值各不相同
要求:空间复杂度 O(n),时间时间复杂度 O(n2)
提示:
1.二叉搜索树是指父亲节点大于左子树中的全部节点,但是小于右子树中的全部节点的树。
2.该题我们约定空树不是二叉搜索树
3.后序遍历是指按照 “左子树-右子树-根节点” 的顺序遍历

思路:两种方法。

  1. 排序最后一个数为根结点。通过和根结点的值做比较,按照二叉搜索树的规则,小的都是左子树,大的都是右子树。对序列进行分割,若在左子树找到比根结点的值大的数,则说明出错了。依次递归下去。
    时间复杂度:O(n^2), n为二叉树节点的个数, 当树为链式时时间复杂度最坏为O(n^2)
class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
        if(sequence.empty()) return false;
        return check(sequence,0,sequence.size()-1);
    }
    bool check(vector<int> sequence,int l,int r){
        if(l>=r) return true;//只有一个数
        int root=sequence[r];
        int j=r-1;
        while(sequence[j]>root&&j>=0) j--;
        for(int i=l;i<=j;i++)
            if(sequence[i]>root) return false;
        return check(sequence,l,j)&&check(sequence,j+1,r-1);
    }
};

空间复杂度:O(n), 当树为链式结构时, 递归深度为n

  • 将后序序列(左右根)通过sort函数变成由小到大的序列,即中序序列(左根右)。可通过栈判断中序序列判断后序序列的排列是否符合左右根,来判断是否是后序遍历。(参考其他同学的)
    时间复杂度:O(nlogn), 时间复杂度取决于排序
    空间复杂度:O(n), 额外数组和辅助栈的空间
class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
        if(sequence.empty()) return false;
        vector<int> midium(sequence);
        sort(midium.begin(),midium.end());
        return check(midium,sequence);
    }
    bool check(vector<int> midium,vector<int> sequence){
        int n=midium.size();
        stack<int> s;
        int i=0,j=0;
        while(i<n){
            s.push(midium[i]);
            while(!s.empty()&&s.top()==sequence[j]){//如果都是左,或者是右和根的交换
                ++j;
                s.pop();
            }
            ++i;
        }
        return j==n;//如果能全部比较完则成功
    }
};

重点:

  • 二叉树很多都可以用递归来思考
  • 通过栈或其他函数改变遍历序列

二叉树中和为某一值的路径(一)

描述
给定一个二叉树root和一个值 sum ,判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n

思路:通过深度遍历递归,如果是叶子结点就检查是否合为sum,否则回溯继续递归。

 bool hasPathSum(TreeNode* root, int sum) {
        if(!root) return false;//如果树为空
        sum-=root->val;
        if(!root->left&&!root->right&&sum==0){//如果是叶结点
            if(total==sum)
                return true;
        } 
        return hasPathSum(root->left,sum)||hasPathSum(root->right,sum);
    }

重点:一开始想通过设置一个全局变量,一点一点加上值计算是否等于sum,但由于有回溯的问题,既要加值又要减值,不如直接用sum减去结点的值来的方便。

二叉树中和为某一值得路径(二)

描述
输入一颗二叉树的根节点root和一个整数expectNumber,找出二叉树中结点值的和为expectNumber的所有路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n

思路:用前序遍历递归,如果是叶子结点且路径和为expectNumber就保存路径。

class Solution {
public:
    using vvi=vector<vector<int>>;
    using vi=vector<int>;
    void dfs(TreeNode* root,int sum,vi &path,vvi &ret){
        path.push_back(root->val);
        if(sum==root->val&&!root->left&&!root->right)
            ret.push_back(path);
        if(root->left) dfs(root->left,sum-root->val,path,ret);
        if(root->right) dfs(root->right,sum-root->val,path,ret);
        path.pop_back();
    }
    vector<vector<int>> FindPath(TreeNode* root,int expectNumber) {
        vvi ret;
        vi path;
        if(!root) return ret;
        dfs(root,expectNumber,path,ret);
        return ret;
    }
};

重点:参考其他同学的,学会用using

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值