給定一個二叉搜索樹，編寫一個函數 kthSmallest 來查找其中第 k 個最小的元素。說明： 你可以假設 k 總是有效的，1 ≤ k ≤ 二叉搜索樹元素個數。示例 1:輸入: root = [3,1,4,null,2], k = 13/ \1 4\2 輸出: 1

解題思路

變量 cnt:統計已經按序遍歷多少個節點了
變量 r： 記錄 第 k 個最小元素的值

根據二叉搜索樹中序遍歷的性質，我們可以按二叉樹節點大小訪問二叉樹，記錄下第k個遍歷的節點，就是我們需要找到的第 k 個最小元素

代碼

/*** Definition for a binary tree node.* public class TreeNode {* int val;* TreeNode left;* TreeNode right;* TreeNode() {}* TreeNode(int val) { this.val = val; }* TreeNode(int val, TreeNode left, TreeNode right) {* this.val = val;* this.left = left;* this.right = right;* }* }*/ class Solution {int r = -1;int cnt = 0;public int kthSmallest(TreeNode root, int k) {mid(root, k);return r;}public void mid(TreeNode root, int k) {if (root == null) return;mid(root.left, k);if (++cnt == k)r = root.val;mid(root.right, k);} }

總結

以上是生活随笔為你收集整理的leetcode230. 二叉搜索树中第K小的元素（中序遍历）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。