题目描述
给定一棵包含 NN 个节点的完全二叉树,树上每个节点都有一个权值,按从 上到下、从左到右的顺序依次是 A_1, A_2, ··· A_NA1,A2,⋅⋅⋅AN,如下图所示:
现在小明要把相同深度的节点的权值加在一起,他想知道哪个深度的节点 权值之和最大?如果有多个深度的权值和同为最大,请你输出其中最小的深度。
注:根的深度是 1。
输入描述
第一行包含一个整数 N(1 \leq N \leq 10^5)N(1≤N≤105)。
第二行包含 N 个整数 A_1, A_2, ··· A_N (−10^5 \leq A_i \leq 10^5)A1,A2,⋅⋅⋅AN(−105≤Ai≤105)。
输出描述
输出一个整数代表答案。
输入输出样例
示例
输入
1. 7 2. 1 6 5 4 3 2 1
输出
2
运行限制
- 最大运行时间:1s
- 最大运行内存: 256M
思路:
题目要求权值最大的节点深度,如果有多个深度的权值和同为最大,输出其中最小的深度。由此可以分成两步,1.求每个深度的权值;2.找到权值最大的节点深度并输出
首先来看第一步,题目给了我们一个串数字,并要求这串数字构成完全二叉树,通过观察,我们发现第1个数在第一层,第2,3个数在第二层,第4,5,6,7个数在第三层,,,,,,以此类推,我们只要找到不同层行的间断点,再将两个间断点之前的所有值相加,就得到了这一层得权值。由上图,观察其A的下标,不难发现层的区间为:2的n次方——2的n+1次方(n=1,2,3.......),确定了区间,我们把区间内的数相加即可
第二步,得到每一层的权值后,我们要做的就是找到权值最大的点,输出其深度
1. p = [] 2. #找到间断点1,2,4,8,16,,,,, 3. for i in range(0, 33): 4. p.append(2 ** i) 5. 6. n = int(input()) 7. a = input().split(" ") 8. 9. ans = [0] * 32 #记录每层的和 10. 11. #遍历所有输入的数 12. for i in range(0, n): 13. #更新每一层的和 14. for j in range(0, len(p)): 15. #在区间内的数相加求和 16. if i + 1 >= p[j] and i + 1 < p[j + 1]: 17. ans[j] += int(a[i]) 18. 19. #找到最大值,输出深度 20. mx, u = -1, -1 21. for i in range(0, len(ans)): 22. if mx < ans[i]: 23. mx, u = ans[i], i 24. print(u + 1)