P8681 [蓝桥杯 2019 省 AB] 完全二叉树的权值

当代码跑的值不全对时，可能是因为部分时间受限，简化一下试试看

#include"bits/stdc++.h"//万能头文件
#include"math.h"
#define long long

using namespace std;

int a[100050],sum[100050]={0},max1=a[0];
int main()
{
	int N,i,j,flag=0;
	cin >> N;
	for(int m=1;m<=N;m++)
	cin >> a[m];
	int p=N,k=0;
/*
  观察发现当根深依次增加时，所对应的节点呈二倍规律
  （1、2、4、8、16...),故可以根据此规律来简化循环次数
（前两次跑就跑过了三个测试点，所以我断定是时间受限，然后想到此方法简化
*/
	while(p)//这里简化一下所循环根深的总值
	{
		p=p/2;
		k++;
	}
	for(i=0;i<k;i++)
	{
		for(j=pow(2,i);j<pow(2,i+1)&&j<=N;j++)
//我们可以发现：每个根深的节点数为2的n次方-2的n-1次方（注意n'的初值）
		{
			sum[i]=sum[i]+a[j];
		}
		max1=max(sum[i],max1);//标记下来最大值
	}
	for(i=0;i<k;i++)
	{
		if(max1==sum[i])
		flag=i;
	}
	cout << flag+1;//flag+1是因为我的sum数组是从sum[0]开始赋的值
	return 0;
}