Leetcode刷题之路-链表,树
剑指offer 36.二叉搜索树与双向链表
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
为了让您更好地理解问题,以下面的二叉搜索树为例:
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
特别地,我们希望可以就地完成转换操作。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。
- 做了一题二叉树与链表结合的题目,没有算法技巧,用STL模板库解决问题
- 首先遍历二叉树,把节点放进vector<Node*>中。
- 因为要排序,所以用sort,自定义仿函数。
- 仿函数auto cmp = [](const Node* v1,const Node* v2){return v1->val < v2 ->val;};
- 指定排序方法
- 接下来就是遍历vector,重构每个节点的左右指针的指向就可以了,这里要分情况讨论。
class Node {
public:
int val;
Node* left;
Node* right;
Node() {}
Node(int _val) {
val = _val;
left = NULL;
right = NULL;
}
Node(int _val, Node* _left, Node* _right) {
val = _val;
left = _left;
right = _right;
}
};
Node* treeToDoublyList(Node* root) {
vector<Node*> box;
auto cmp = [](const Node* v1,const Node* v2){return v1->val < v2->val;};
if(root == NULL) return root;
box.push_back(root);
dfs(box,root);
sort(box.begin(),box.end(),cmp);
for(int i=0,len=box.size();i<len;i++){
if(i==0&&len>1){
box[i]->left = box[len-1];
box[i]->right = box[i+1];
}else if(i==0&&len==1){
box[0]->left = box[0];
box[0]->right = box[0];
}else if(i==len-1){
box[i]->left = box[i-1];
box[i]->right = box[0];
}else{
box[i]->left = box[i-1];
box[i]->right = box[i+1];
}
}
return box[0];
}
void dfs(vector<Node*>& box,Node* root){
if(!root->left && !root->right){
return;
}
if(root->left){
box.push_back(root->left);
dfs(box,root->left);
}
if(root->right){
box.push_back(root->right);
dfs(box,root->right);
}
}