LeetCode 54.二叉搜索树的第k大节点

给定一棵二叉搜索树,请找出其中第k大的节点。

示例 1:

输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4

tips: 逆中序遍历, 维护一个全局计数器,当计数器到达k时,记录topk值。

int num=0;
int topk=0;
void postVisit(struct TreeNode* root,int k);
int kthLargest(struct TreeNode* root, int k){
	num=0;
	postVisit(root,k);
	return topk;
}

void postVisit(struct TreeNode* root,int k) {
	if(root==NULL) {
		return;
	}
	postVisit(root->right,k);
	num++;
	if(num==k) {
        topk = root->val;
        return;
    }
	postVisit(root->left,k);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章