230. 이진 검색 트리에서 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 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);
}
}