1.问题描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
节点的结构体:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
2.分析
根据二叉搜素的特性,如果是中序遍历,那么遍历的结果就是从小到大的排序。
方法一:可以使用STL中的queue,一次中序遍历,把节点从小到大放到queue中。
之后从queue中提出每一个节点,改变节点的指针方向就可以了。
方法二:可以在一次中序遍历,直接改变节点的指针方向。
3.代码
//使用queue方法,获得我们需要的queue<TreeNode*>
void getOrderNode(queue<TreeNode*>& nodes,TreeNode* pRootOfTree)
{
if(pRootOfTree)
{
getOrderNode(nodes,pRootOfTree->left);
nodes.push(pRootOfTree);
getOrderNode(nodes,pRootOfTree->right);
}
}
//直接排序,这里之所以使用TreeNode*& ,是为了改变pre
void ConvertNode(TreeNode* pRootOfTree,TreeNode*& pre)
{
if(pRootOfTree)
{
ConvertNode(pRootOfTree->left,pre);
pRootOfTree->left = pre;
if(pre) pre->right = pRootOfTree;
pre = pRootOfTree;
ConvertNode(pRootOfTree->right,pre);
}
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree == nullptr)
return pRootOfTree;
/*
这里是方法一的使用,这里注释了
queue<TreeNode*> nodes;
getOrderNode(nodes,pRootOfTree);
TreeNode* pHead = nodes.front();
pHead->left = nullptr;
TreeNode* pCur = pHead;
nodes.pop();
while(nodes.size())
{
TreeNode* pNode = nodes.front();
nodes.pop();
pNode->left = pCur;
pCur->right = pNode;
pCur = pCur->right;
}
return pHead;
*/
TreeNode* pre = nullptr;
ConvertNode(pRootOfTree,pre);
TreeNode* pHead = pRootOfTree;
//因为pRootOfTree,pre两个节点都不是头结点,需要向左移动到头结点
while(pHead->left)
{
pHead = pHead->left;
}
return pHead;
}