【Python】2019年蓝桥杯省赛真题——完全二叉树的权值

蓝桥杯 2019 省 A&B:完全二叉树的权值

题目描述

给定一棵包含 N N N 个节点的完全二叉树,树上每个节点都有一个权值,按从上到下、从左到右的顺序依次是 A 1 , A 2 , ⋯ A N A_1,A_2, \cdots A_N A1,A2,⋯AN,如下图所示:

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

注:根的深度是 1 1 1。

输入描述

第一行包含一个整数 N ( 1 ≤ N ≤ 1 0 5 ) N (1\le N \le 10^5) N(1≤N≤105)。

第二行包含 N N N 个整数 A 1 , A 2 , ⋯   , A N ( − 1 0 5 ≤ A i ≤ 1 0 5 ) A_1,A_2, \cdots, A_N (-10^5\le A_i \le 10^5) A1,A2,⋯,AN(−105≤Ai≤105)。

输出描述

输出一个整数代表答案。

样例 #1

样例输入 #1

复制代码
7
1 6 5 4 3 2 1

样例输出 #1

复制代码
2

运行限制

  • 最大运行时间:1s
  • 最大运行内存:256M

代码实现

python 复制代码
n = int(input())
tree = list(map(int, input().split()))
values = []
for i in range(len(bin(n)[2:])):	# 二叉树深度
    values.append(sum(tree[2**i-1: min(2**(i+1)-1, len(tree))]))
print(values.index(max(values)) + 1)
相关推荐
天选之女wow4 分钟前
【代码随想录算法训练营——Day32】动态规划——509.斐波那契数、70.爬楼梯、746.使用最小花费爬楼梯
算法·leetcode·动态规划
java1234_小锋7 分钟前
TensorFlow2 Python深度学习 - TensorFlow2框架入门 - 神经网络基础原理
python·深度学习·tensorflow·tensorflow2
JJJJ_iii8 分钟前
【深度学习03】神经网络基本骨架、卷积、池化、非线性激活、线性层、搭建网络
网络·人工智能·pytorch·笔记·python·深度学习·神经网络
红衣小蛇妖11 分钟前
LeetCode-704-二分查找
java·算法·leetcode·职场和发展
rongqing201916 分钟前
问题记录:一个简单的字符串正则匹配算法引发的 CPU 告警
算法
JJJJ_iii17 分钟前
【深度学习05】PyTorch:完整的模型训练套路
人工智能·pytorch·python·深度学习
无限进步_33 分钟前
C语言字符串与内存操作函数完全指南
c语言·c++·算法
rengang6641 分钟前
07-逻辑回归:分析用于分类问题的逻辑回归模型及其数学原理
人工智能·算法·机器学习·分类·逻辑回归
Zzzzmo_1 小时前
【Java】杨辉三角、洗牌算法
java·数据结构·算法
闻缺陷则喜何志丹1 小时前
【C++贪心】P10537 [APIO2024] 九月|普及+
c++·算法·贪心·洛谷