给定一个二叉搜索树的根节点 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;}
};