230. 二叉搜索树中第K小的元素
解题思路
- 改造中序遍历算法
- 针对中序遍历 每次遍历到一个节点的时候 rank++ 表示当前遍历元素的顺序 同时记录当前遍历节点的值
/*** 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 res = 0;int rank = 0;public int kthSmallest(TreeNode root, int k) {traverse(root,k);return res;}public void traverse(TreeNode root,int k){if(root == null){return;}traverse(root.left,k);// 改造中序遍历算法rank++;if(k == rank){// 找到第k小的元素res = root.val;return;}traverse(root.right,k);}
}