給定一個二叉搜索樹,編寫一個函數 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);}
}