某公司组织架构以二叉搜索树形式记录,节点值为处于该职位的员工编号。请返回第 cnt
大的员工编号。
示例 1:
输入:root = [7, 3, 9, 1, 5], cnt = 2
7
/ \
3 9
/ \
1 5
输出:7
示例 2:
输入: root = [10, 5, 15, 2, 7, null, 20, 1, null, 6, 8], cnt = 4
10
/ \
5 15
/ \ \
2 7 20
/ / \
1 6 8
输出: 8
提示:
- 1 ≤ cnt ≤ 二叉搜索树元素个数
思路:
1、二叉搜索树的中序遍历(左根右)为递增序列,我们利用(右根左)即可得到递减序列。先递归至最右结点即最大结点处,利用cnt--是否为0判断当前结点值大小次序是否符合要求,不符合则继续递归当前结点左子树。
2、递归:
终止条件 结点为空或已经找到目标结点 递归右子树 一直到最大结点处 条件判断 1、cnt--
2、判断cnt是否为0,是则赋值res,直接返回
3、不是则继续递归左子树
返回值 返回res
代码实现:
/**
* 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 findTargetNode(TreeNode* root, int cnt) {
findNode(root, cnt);
return res;
}
void findNode(TreeNode* root, int& cnt) //cnt需要引用!!!
{
if(!root || res != 0) //结点为空或者res已被赋值即已经找到目标值
return;
findNode(root->right, cnt); //递归至最大结点处
--cnt;
if(cnt == 0)
{
res = root->val;
return;
}
findNode(root->left, cnt);
}
private:
int res = 0;
};