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);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章