题目详见https://leetcode.cn/problems/kth-largest-sum-in-a-binary-tree/
读完最开始的题目描述一直无感,直到读到了最后那句:
如果两个节点与根节点的距离相同,则认为它们在同一层
恍然大悟原来是广度优先,就类似走一个金字塔形状的迷宫从塔尖开始走,走到最后也就遍历到了所有的结点,在这个过程中也就记录了需要的信息。
注释代码
/**
* 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:
long long kthLargestLevelSum(TreeNode* root, int k) {
queue<TreeNode* > q;
q.push(root);
vector<long long> levelSumArray;
while(!q.empty()){
long long levelSum = 0, size = q.size();
for(int i = 0; i < size; i++){
// 补充一下Queue的全部操作
// q.empty() 如果队列为空返回true,否则返回false
// q.size() 返回队列中元素的个数
// q.pop() 删除队列首元素但不返回其值
// q.front() 返回队首元素的值,但不删除该元素
// q.push(x) 在队尾压入新元素
// q.back() 返回队列尾元素的值,但不删除该元素
TreeNode *node = q.front();
q.pop();
levelSum += node->val;
if(node->left){
// 塞入下一层左子树的根节点
q.push(node->left);
}
if(node->right){
// 塞入下一层右子树的根节点
q.push(node->right);
}
}
levelSumArray.push_back(levelSum);
}
if(levelSumArray.size() < k){ // 题目描述中提到过
return -1;
}
sort(levelSumArray.begin(), levelSumArray.end());
return *(levelSumArray.end() - k); // 排序后取倒数第K个,也就是第K大的。括号里面时位置也就是地址,然后*取值。
}
};