897.递增顺序搜索树

61 阅读1分钟

题目:
给你一棵二叉搜索树的 root ,请你 按中序遍历 将其重新排列为一棵递增顺序搜索树,使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。
算法:
这个题目需要increasingBST返回的节点是最左侧的节点,由要将左节点的right节点指向当前节点。考虑inorder用两个返回值。 实际上,中序遍历,记录上次访问的node,和dummynode就可以解决

func increasingBST(root *TreeNode) *TreeNode {
	dummyNode := &TreeNode{}
	lastNode := dummyNode
	var inorder func(node *TreeNode) 
	inorder = func(node *TreeNode)  {
		if node == nil {
			return 
		}
		inorder(node.Left)
		lastNode.Right = node
		node.Left = nil
		lastNode = node
		inorder(node.Right)
	}
	inorder(root)
	
	return dummyNode.Right
}