2020-07-31:给定一个二叉搜索树(BST),找到树中第K 小的节点。

福哥答案2020-07-31:

BST 的中序遍历是升序序列。
1.递归法。
时间复杂度:O(N),遍历了整个树。
空间复杂度:O(N),用了一个数组存储中序序列。
2.迭代法。
时间复杂度:O(H+k),其中 H 指的是树的高度,由于我们开始遍历之前,要先向下达到叶,当树是一个平衡树时:复杂度为 O(logN+k)。当树是一个不平衡树时:复杂度为 O(N+k),此时所有的节点都在左子树。
空间复杂度:O(H+k)。当树是一个平衡树时:O(logN+k)。当树是一个非平衡树时:O(N+k)。

golang代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package test30_kth
  
import (
    "fmt"
    "testing"
)
  
//Definition for a binary tree node.
type TreeNode struct {
    Val   int
    Left  *TreeNode
    Right *TreeNode
}
  
//BST 的中序遍历是升序序列
//go test -v -test.run TestKth
func TestKth(t *testing.T) {
    root := &TreeNode{}
    root.Val = 3
    root.Right = &TreeNode{}
    root.Right.Val = 4
    root.Left = &TreeNode{}
    root.Left.Val = 1
    root.Left.Right = &TreeNode{}
    root.Left.Right.Val = 2
    ret := kthSmallest1(root, 2)
    fmt.Println("递归法:", ret)
    ret = kthSmallest2(root, 2)
    fmt.Println("迭代法:", ret)
}
  
//递归法
//时间复杂度:O(N),遍历了整个树。
//空间复杂度:O(N),用了一个数组存储中序序列。
func kthSmallest1(root *TreeNode, k int) int {
    nums := inorder(root, &[]int{})
    return (*nums)[k-1]
}
func inorder(root *TreeNode, arr *[]int) *[]int {
    if root == nil {
        return arr
    }
    inorder(root.Left, arr)       //左
    *arr = append(*arr, root.Val) //根
    inorder(root.Right, arr)      //右
    return arr
}
  
//迭代法
//时间复杂度:O(H+k),其中 H 指的是树的高度,由于我们开始遍历之前,要先向下达到叶,当树是一个平衡树时:复杂度为 O(logN+k)。当树是一个不平衡树时:复杂度为 O(N+k),此时所有的节点都在左子树。
//空间复杂度:O(H+k)。当树是一个平衡树时:O(logN+k)。当树是一个非平衡树时:O(N+k)。
func kthSmallest2(root *TreeNode, k int) int {
    stack := make([]*TreeNode, 0)
    for {
        for root != nil {
            //push
            stack = append(stack, root)
  
            root = root.Left
        }
  
        //pop
        root = stack[len(stack)-1]
        stack = stack[0 : len(stack)-1]
  
        k = k - 1
        if k == 0 {
            return root.Val
        }
        root = root.Right
    }
}

  敲 go test -v -test.run TestKth 命令,结果如下:

 

posted @   福大大架构师每日一题  阅读(300)  评论(0编辑  收藏  举报
编辑推荐:
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· AI 智能体引爆开源社区「GitHub 热点速览」
· Manus的开源复刻OpenManus初探
· 写一个简单的SQL生成工具
点击右上角即可分享
微信分享提示