二叉搜索树与双向链表
- 参与人数:1237时间限制:1秒空间限制:32768K
- 通过比例:24.74%
- 最佳记录:0 ms|8552K(来自 牛客456244号)
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
主要思路:
中序遍历,遍历的顺序就是,数值从小到大
用一个指针指向,当前指针的前一个节点,然后改变left和right的指向。
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(!pRootOfTree){
return NULL;
}
stack<TreeNode* > sta;
TreeNode *pre=NULL;
TreeNode *current=pRootOfTree;
while(current||!sta.empty()){
if(current){
sta.push(current);
current=current->left;
}else{
current=sta.top();
if(pre==NULL){
pRootOfTree=current;
pre=current;
}else{
pre->right=current;
current->left=pre;
pre=current;
}
sta.pop();
current=current->right;
}
}
return pRootOfTree;
}