题目:
从左向右遍历一个数组,通过不断将其中的元素插入树中可以逐步地生成一棵二叉搜索树。
给定一个由不同节点组成的二叉搜索树 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;
}
};