蓝桥杯—二叉树的权值（C语言解法）

题目描述

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

现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点 权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。
注：根的深度是 1。

输入描述

第一行包含一个整数 N。
第二行包含 N 个整数 A1, A2, · · · AN 。

输出描述

输出一个整数代表答案

输入输出样例

输入

7
1 6 5 4 3 2 1

输出

2

思路

1. 因为是二叉树，所以每层最大节点数是确定的，i层的最大节点数为 2^(i-1)
2. 将二叉树各节点的权值存入数组中，对该数组进行遍历，每层的节点数从1开始，如大于最大节点数，则层数加1（层数初始值为1）
3. 将根节点作为max的初始值，若某层权值和大于max，则level为该层层数（level为最终结果）
4. 最后一层要单独比较，因为最后一层的节点数是小于等于最大值的，层数无法增加，权值和也不能在循环中与max比较
5. 先增加层数，在比较max与上一层权值和原因：count（层级数）的初始值为1，而进入循环后是从第二层开始的（max的值设为了根节点的值），所以count落后1，要先加1

#include<stdio.h>
#include<math.h>
int main(){
	int max,num,level=1,count=1,nums=0;
	int snum[100000];
	int i,j=1;
	scanf("%d",&num);
	for(i=0;i<num;i++){
		scanf("%d",&snum[i]);
	}//读入数据并存储至数组
	max=snum[0];
	for(i=1;i<num;i++,j++){//遍历数组
		if(j>pow(2.0,count)){//新一层开始
			count++;//层数加1
			if(max<nums){max=nums;level=count;}//比较max与上一层节点和的大小
			j=1;//该层节点数从1开始
			nums=snum[i];//该层权值和为该层第一个节点的值
		}
		else{nums+=snum[i];}//每层除第一个节点之外的节点直接加到该层权值和中
	}
	if(max<nums){max=nums;level=count+1;}//最后一层的权值和
	printf("%d",level);
	return 0;
}