Leetcode-2583-二叉树中的第 K 大层和-c++

题目详见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大的。括号里面时位置也就是地址，然后*取值。
    }
};