给定一个二叉搜索树的根节点 root
,和一个整数 k
,请你设计一个算法查找其中第 k
个最小元素(从 1 开始计数)。
示例 1:输入:root = [3,1,4,null,2], k = 1 输出:1
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int value=0;
int cnt=0;
void dfs(TreeNode* node,int k){
if(!node){return;}
dfs(node->left,k);
cnt++;
if(cnt==k){value=node->val;}//二叉搜索树中序遍历即为顺序遍历,返回第k个节点的值
dfs(node->right,k);
}
int kthSmallest(TreeNode* root, int k) {
int cnt=0;
dfs(root,k);
return value;
}
};