题目:
给你一棵二叉树的根节点 root 和一个正整数 k 。
树中的 层和 是指 同一层 上节点值的总和。
返回树中第 k 大的层和(不一定不同)。如果树少于 k 层,则返回 -1 。
注意,如果两个节点与根节点的距离相同,则认为它们在同一层。
示例1:
输入:root = [5,8,9,2,1,3,7,4,6], k = 2
输出:13
解释:树中每一层的层和分别是:
- Level 1: 5
- Level 2: 8 + 9 = 17
- Level 3: 2 + 1 + 3 + 7 = 13
- Level 4: 4 + 6 = 10
第 2 大的层和等于 13 。
示例 2:
输入:root = [1,2,null,3], k = 1
输出:3
解释:最大的层和是 3 。
提示:
树中的节点数为 n
2 <= n <= 10^5
1 <= Node.val <= 10^6
1 <= k <= n
java代码 :
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public long kthLargestLevelSum(TreeNode root, int k) {
if(root == null) return -1;//根节点为空,直接返回-1。
ArrayList<Long> arr = new ArrayList<>();//暂存层和,也即每层中节点value的累加值。
Deque<TreeNode> que = new ArrayDeque<>();//队列。
que.addFirst(root);//先加入根节点。
while(que.isEmpty() == false) {
int num = que.size();//当前层中的节点数。
long sum = 0;//累加值。
for(int i = 0; i <num; i++) {//每个节点分别出队累加并在队列中加入新节点。
TreeNode temp = que.pollLast();
sum += temp.val;
if(temp.left != null) que.addFirst(temp.left);//注意非空时才能累加。
if(temp.right != null) que.addFirst(temp.right);
}
arr.add(sum);//当前层和暂存。
}
arr.sort(Comparator.naturalOrder());//排序。
if(arr.size() >= k) return arr.get(arr.size() - k);//若层数比k大,输出第k大。
else return -1;
}
}