本文旨在对于个人知识的梳理以及知识的分享,如果有不足的地方,欢迎大家在评论区指出
题目描述
二叉树数据结构TreeNode可用来表示单向链表(其中left置空,right为下一个链表节点)。实现一个方法,把二叉搜索树转换为单向链表,要求依然符合二叉搜索树的性质,转换操作应是原址的,也就是在原始的二叉搜索树上直接修改。
返回转换后的单向链表的头节点。
注意:本题相对原题稍作改动
示例:
输入: [4,2,5,1,3,null,6,0]
输出: [0,null,1,null,2,null,3,null,4,null,5,null,6]
提示:
- 节点数量不会超过 100000。
题目链接
题目分析
题目中所给的是一颗二叉搜索树,我们可以发现转化成的单向链表是有序的,也就是该二叉搜索树的中序遍历,这里可以用一个简单的案例模拟一下该过程:
所对应的操作也就是tail.right = root; tail = tail.right; root.left=null
,所以我们只需要中序遍历该二叉搜索树,然后执行上面代码的操作就可以将该二叉搜索树转化为一个单向链表
解题思路
Python
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def convertBiNode(self, root: TreeNode) -> TreeNode:
head, tail = None, None
def dfs(root):
nonlocal tail, head
if not root:
return
dfs(root.left)
if not tail:
head = root
tail = head
else:
tail.right = root
tail = tail.right
root.left = None
dfs(root.right)
dfs(root)
return head
Java
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private TreeNode head, tail;
private void dfs(TreeNode root){
if(root == null) return;
dfs(root.left);
if(tail == null){
head = root;
tail = head;
}else{
tail.right = root;
tail = tail.right;
}
root.left = null;
dfs(root.right);
}
public TreeNode convertBiNode(TreeNode root) {
dfs(root);
return head;
}
}