題目:二叉搜索樹和雙向鏈表的轉化
- 二叉搜索樹裏面每個節點有兩個指針,分別指向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;
}
};