题目:二叉搜索树和双向链表的转化
- 二叉搜索树里面每个节点有两个指针,分别指向left 和 right。
- 双向链表内的每个节点也有两个指针,分别指向left 和 right。
- 所以,两者是可以相互转化的。
- 我们知道二叉搜索树的中序序列有序的。所以我们基于中序序列来将其转化为双向链表。
- 对于每一个节点来说,它的left是它中序遍历的上一个元素,right是中序遍历的下一个元素。
- 所以在中序遍历的过程中,我们简历一个lastNode变量,让其维持中序遍历访问的上一个元素。
- 则lastNode->right = current Node, currentNode->left = lastNode
- 通过这样的方式可以将二叉搜素树转化为双向链表
- 代码如下:
#include <iostream>
#include <vector>
#include <string>
#include <cstring>
using namespace std;
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
class Solution {
public:
void ConvertBase(TreeNode* root, TreeNode** lastNode){
if(root == NULL)
return NULL;
if(root->left){
ConvertBase(root->left, lastNode);
}
root->left = *lastNode;
if(*lastNode){
(*lastNode)->right = root;
}
(*lastNode) = root;
if(root->right){
ConvertBase(root->right, lastNode);
}
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if(pRootOfTree == NULL)
return NULL;
TreeNode* lastNode = NULL;
ConvertBase(pRootOfTree, &lastNode);
TreeNode* res = pRootOfTree;
while (res && res->left){
res = res->left;
}
return res;
}
};