蓝桥 完全二叉树的权值 树 层序遍历 双指针 java

完全二叉树的权值

在这里插入图片描述
⭐ 层序遍历:枚举每层 起点 + 节点数,细节迭代层数,细节开大数组
⭐ 注意 int 的范围,注意下标从 0 或 1 开始
⭐ 2 的 d 次方:1 << d
⭐ O(n)

package algorithm.lanQiao.宽搜;

import java.io.*;
import java.util.*;

public class 完全二叉树的权值
{
	static int N = 100010, INF = 0x3f3f3f3f;
	static int[] a = new int[2 * N];

	public static void main(String[] args)
	{
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();

//		for (int i = 0; i < n; i++)
		for (int i = 1; i <= n; i++)
			a[i] = sc.nextInt();

		int d = 1;
		int md = 1;
		long max = -INF;
		// i 是每一层的起点,j 是每一层的步长,由于数组开大了不少,无需判断边界问题
//		for (int i = 0; i < n; i += (1 << (d - 1)), d++)
		for (int i = 1; i < n; i *= 2, d++)
		{
			long sum = 0;
			for (int j = 0; j < (1 << (d - 1)); j++)
			{
				sum += a[i + j];
			}
			if (sum > max)
			{
//				System.out.println(sum);
				md = d;
				max = sum;
			}
//			d++; //错误示例,得先跳了一层在 ++
		}

		System.out.println(md);
//		System.out.println(max);

	}
}

posted @ 2023-03-31 10:04  兑生  阅读(19)  评论(0)    收藏  举报  来源
Live2D