/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param proot TreeNode类 * @param k int整型 * @return int整型 */ int KthNode(TreeNode* proot, int k) { int count = 0; return KthNode2(proot, k, count); } int KthNode2(TreeNode *proot, int k, int &count) { if (!proot) { return -1; } int l = KthNode2(proot->left, k, count); if (l != -1) { return l; } ++count; if (count == k) { return proot->val; } return KthNode2(proot->right, k, count); } };
思路:二叉搜索树中序遍历即为顺序遍历,所以中序遍历并计数即可。