每天一道leetcode:剑指 Offer 36. 二叉搜索树与双向链表(中等深度优先遍历递归)
今日份题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
示例
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
题目思路
根据二叉搜索树的性质,我们知道,中序遍历可以得到树节点的从小到大的排序,那我们就在中序递归遍历的基础上改变链表。使用pre节点记录上个节点的位置,使用cur节点记录当前节点的位置,使用head节点记录链表头部的位置,然后每次让pre节点和cur节点互相指,pre左cur右,最后让head和pre节点相互指就能得到结果链表了。具体看代码注释。
补充:中序遍历
树的一种遍历方法,遍历顺序是左->根->右,如果左或右是树而非节点,那么就在子树中继续左根右,最后递归得到顺序。
int inOrder(Node* root)
{if(root->left) inOrder(root->left);return root->val;if(root->right) inOrder(root->right);
}
代码
/*
// Definition for a Node.
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;}
};
*/
class Solution
{
public:Node *pre,*head;void dfs(Node* cur) {if(cur==NULL) return;//中序遍历dfs(cur->left);//左边pre,右边cur,二者相连,pre继续向后走直到结束if(pre!=NULL) pre->right=cur;else head=cur;cur->left=pre;pre=cur;dfs(cur->right);}Node* treeToDoublyList(Node* root) {if(root==NULL) return NULL;dfs(root);//中间连好了,头尾相连head->left=pre;pre->right=head;return head;}
};
提交结果
欢迎大家在评论区讨论,如有不懂的部分,欢迎在评论区留言!
更新不易,宝子们点个赞支持下,谢谢!