【LeetCode & 剑指offer刷题】树题14：36 二叉搜索树与双向链表
【LeetCode & 剑指offer 刷题笔记】目录（持续更新中...）
36 二叉搜索树与双向链表
题目描述
输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
/*
方法一：中序遍历迭代法
参考题目： Validate Binary Search Tree）
*/
class Solution
{
public
:
TreeNode
*
Convert
(
TreeNode
*
root
)
{
if
(
root
==
nullptr
)
return
nullptr
;
TreeNode
*
head
=
root
;
while(head->left) head = head->left;
//
二叉搜索树中最左子结点即为中序遍历的第一个结点
stack
<
TreeNode
*>
s
;
TreeNode
*
p
=
root
,
*
pre
=
nullptr
;
while
(
p
||
!
s
.
empty
())
{
if
(
p
)
{
s
.
push
(
p
);
p
=
p
->
left
;
}
else
{
p
=
s
.
top
();
p->left = pre; //连接p与pre
if(pre) pre->right = p; //连接pre与p
pre = p; //更新pre
s
.
pop
();
p
=
p
->
right
;
}
}
return head;
}
};
/*
方法二：中序遍历递归法（左、根、右）
*/
class
Solution
{
private
:
TreeNode
*
pre
;
public
:
TreeNode
*
Convert
(
TreeNode
*
pRootOfTree
)
{
if
(
pRootOfTree
==
nullptr
)
return
nullptr
;
pre
=
nullptr
;
//初始化成员变量,用于记录上一次访问的结点
convertNode
(
pRootOfTree
);
TreeNode
*
head
=
pRootOfTree
;
while
(
head
->
left
)
head
=
head
->
left
;
//二叉搜索树中最左子结点即为中序遍历的第一个结点
return
head
;
}
private
:
void
convertNode
(
TreeNode
*
cur
)
{
if
(
cur
==
nullptr
)
return
;
//转换左子树
convertNode
(
cur
->
left
);
//转换根结点
cur
->
left
=
pre
;
if
(
pre
)
pre
->
right
=
cur
;
pre
=
cur
;
//记录被处理过的结点
//转换右子树
convertNode
(
cur
->
right
);
}
};
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
/*
中序遍历迭代法
*/
posted @
2019-01-05 19:49 wikiwen 阅读(
...) 评论(
...) 编辑 收藏