题目描述
给定一棵包含 N 个节点的完全二叉树,树上每个节点都有一个权值,
按从 上到下、从左到右的顺序依次是 A1, A2, · · · AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,
他想知道哪个深度的节点 权值之和最大?
如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是 1。
输入
第一行包含一个整数 N。 第二行包含N个整数A1,A2,··· AN。
输出
输出一个整数代表答案。
样例输入
7
1 6 5 4 3 2 1
样例输出
2
代码:
#include <iostream>
#include <algorithm>
using namespace std;
int s[1005],a[1005];
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
int n,num;
cin>>n;
int k=1,flag=0,temp=1;
/* k 数组的下标变量,即每个深度的权值
都被加和后放在数组s的同一下标内*/
/* flag 表示已被放入数组s内的个数
temp表示每个深度的节点个数*/
while(n--)
{
cin>>num;
if(flag<temp)
{
s[k]+=num;
a[k]+=num;
/*另定义了一个与数组s相同的数组a
便于找出最大权和的(最小)深度*/
flag++;
}
if(flag==temp)
{
flag=0;
temp*=2;
k++;
}
}
sort(a,a+1005,cmp);
/*对数组a进行排序,找出最大值*/
int ans;
for(int i=1;;i++)
{
if(s[i]==a[0])
{
ans=i;
break;
}
}
cout<<ans;
return 0;
}
- 这个是我自己写的,能过的数据样例应该很少。下附一份学长推荐的博主的AC代码,点此跳转。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000000;
int N, a[MAXN + 1];
int main() {
scanf("%d", &N);
for (int i = 1; i <= N; ++i)
scanf("%d", a + i);
long long m = -MAXN;
int ans = 0;
for (int i = 1; (1 << (i - 1)) < N; ++i) {
long long t = 0;
for (int j = 1 << (i - 1); j < (1 << i) && j < N; ++j)
t += a[j];
if (m < t) {
m = t;
ans = i;
}
}
cout << ans << endl;
return 0;
}
总结
- 虽然过了题,可能是因为数据样题不全。
- 第二份代码看不懂,还是得继续加油鸭!