[蓝桥杯2019初赛]完全二叉树的权值

给定一棵包含N 个节点的完全二叉树，树上每个节点都有一个权值，按从
上到下、从左到右的顺序依次是A1, A2, AN，如下图所示：
无图嘻嘻

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点
权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。
注：根的深度是1。
输入
第一行包含一个整数N。
第二行包含N 个整数A1, A2, AN
对于所有评测用例，1<=N<=100000, -100000<=Ai<=100000。
输出
输出一个整数代表答案。
样例输入 Copy
7
1 6 5 4 3 2 1
样例输出 Copy
2
根据完全二叉树深度为：log2 (n+1);
因为不一定是满二叉树
所以把最后一层单独拿出来写即可

#include<algorithm>
#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n;
	cin >> n;
	long long a[100005];
	long long ans=0;
	int depth = log(n + 1) / log(2);
	if (pow(2, depth) < n)
	{
		depth += 1;
	}
	//cout << depth << endl;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	int cen=1;
	int t = 0;
	for (int i = 1; i < depth; i++)
	{
		long long p = 0;
		for (int j =pow(2,t); j < pow(2,t)+pow(2,t); j++)
		{
			p += a[j];
		}
		//cout << p << endl;
		if (ans < p)
		{
			ans = p;
			cen = i;
		}
		t++;
		//cout << t << endl;
	}
	long long np=0;
	for (int j = pow(2, depth - 1); j <= n; j++)
	{
		np += a[j];
	}
	if (ans < np)
	{
		cen = depth;
	}
	cout << cen << endl;
}