1240. 完全二叉树的权值 Java题解 （双指针）【第十届蓝桥杯省赛C++A/B组,JAVA A组】

输入样例：

7
1 6 5 4 3 2 1

输出样例：

2

---

解题思路：

设 i 完全二叉树的第 i 层，则第 i 层的开始下标为 2^(i - 1)，第 i 层的结束下标为 2^ i - 1。

通过双层循环，外层控制二叉树的层数，内层计算每一层的和，求最大值。时间复杂度为O(n).

Java代码：

import java.io.*;

public class Main {
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		int []arr = new int[n + 1];
		String[] split = br.readLine().split(" ");
		for(int i = 1; i <= n; i++)
			arr[i] = Integer.parseInt(split[i - 1]);
		
		long max = Long.MIN_VALUE;  //最大和
		int ans = 1;  //最大层数
		for(int deep = 1, i = 1; i <= n; i *= 2, deep++) {  //deep为当前层数，i为该层的开始下标
			long sum = 0;
			for(int j = i; j < (1 << deep) && j <= n; j++) {  //遍历每层
				sum += arr[j];
			}
			if(sum > max) {
				max = sum;
				ans = deep;
			}
		}
		System.out.println(ans);
	}
}