二叉搜索树序列(利用队列实现)

题目:

从左向右遍历一个数组,通过不断将其中的元素插入树中可以逐步地生成一棵二叉搜索树。

给定一个由不同节点组成的二叉搜索树 root,输出所有可能生成此树的数组。

示例:

输入: root = [2,1,3]
输出: [[2,1,3],[2,3,1]]
解释: 数组 [2,1,3]、[2,3,1] 均可以通过从左向右遍历元素插入树中形成以下二叉搜索树
       2 
      / \ 
     1   3

解题思路:

 本题相当于考查遍历这棵树的所有方式。

首先,父节点的优先级一定是最高的,其左右节点的优先级是相同的,当左节点被访问后,它的左右子节点和上一层的右节点优先访问级别也是相同的。

所以也可以看作是对同一优先级的所有节点的全排列,这里我们利用队列来保存需要全排列的所有节点

既然是全排列,就肯定是要用到递归算法,在递归的过程中因为需要回溯,所以这里的队列我们采用双向队列,便于尾删。

源代码如下:

//相当于遍历这棵树的所有方式
class Solution {
public:
    vector<vector<int>> res;
    vector<int> temp;
    void dfs(deque<TreeNode*>& dq)
    {
        if(dq.empty())
        {
            //当候选队列为空时,说明已经遍历完了二叉搜索树,此时将temp添加到res中
            res.push_back(temp);
            return;
        }
        int len=dq.size();
        for(int i=0;i<len;i++)
        {
            TreeNode* node=dq.front();//保存选择的这个节点
            dq.pop_front();
            //队头元素添加到临时数组中
            temp.push_back(node->val);
            //每选择一个节点之后,都要在候选队列中添加其左右孩子
            if(node->left) dq.push_back(node->left);
            if(node->right) dq.push_back(node->right);
            dfs(dq);
			//回溯
            //先在队列尾部删除上一步加入的子节点
            if(node->left) dq.pop_back();
            if(node->right) dq.pop_back();
            //要将当前节点还原到候选队列中
            dq.push_back(node);
            //从临时数组中删除
            temp.pop_back();
        }
    }
    vector<vector<int>> BSTSequences(TreeNode* root) {
        if(root==NULL) return {{}};
        deque<TreeNode*> dq;//候选队列
        //数组的首元素肯定是根节点
        dq.push_back(root);
        dfs(dq);
        return res;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值