Q:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
- 二叉搜索树本身就是一棵有序的树,要转换成一个排序的链表,则就是它的中序遍历,遍历的第一个结点是最左结点
- 定义一个prev指针保存它的前一个结点,调整指针指向
画图:
|
代码:
定义树的结点:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
排序:
class Solution {
public:
void _Convert(TreeNode* cur,TreeNode*& prev)
{
if(cur==NULL)
return ;
_Convert(cur->left,prev);
cur->left=prev;
if(prev!=NULL) //防止让空指针指向结点
prev->right=cur;
prev=cur;
_Convert(cur->right,prev);
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
TreeNode* prev=NULL;
TreeNode* cur=pRootOfTree;
_Convert(cur,prev);
while(cur&&cur->left)
{
cur=cur->left; //找到最左节点为双向链表的头节点
}
return cur;
}
};