leetcode 230. 二叉搜索树中第 K 小的元素

题目如下

数据范围

利用一个全局的计数器每过一个节点就加一，再利用中序遍历的特性（即遍历到的节点是从小到大的）

通过代码

/*** 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 count = 1;int ans = -1;void in(TreeNode* root) {if (root == nullptr || ans != -1)return;in(root->left);if(ans != -1)return;if (count == 1) {ans = root->val;return;}count--;in(root->right);}int kthSmallest(TreeNode* root, int k) {count = k;in(root);return ans;}
};