备战蓝桥杯之二叉树初赛大题最后一天

先上题目，写个程序确认刷的到位了。，这里我按代码的角度写博文

初步分析:题目问的是深度为多少时，权值最大，这里我先假设深度为n,那么怎么求他的权值和呢？一个深度一个权值和，两个深度两个权值和，所以想到用二维数组来存储，索引就是他的深度

第二步，定义多大的二维数组？
经过我的探索，数组索引多大取决二叉树的深度，二叉树的深度有性质4，题目会给节点的个数，则深度k=（log2 n)+1

这里我先做个小测试，导入math,math可以直接调用对数，题目给N的个数是100000，算出来最多有16层。

然后第三步，如何吧每层的值存入二维数组？如何确定每层的值？
确定每层的值，我想到for遍历，比如二维数组索引为1，存储第一层的全部值，索引为2，存储第二层的全部值，我想到了append方法，遍历一次，存进一个值，用(index,index+pow(2,i-1)确定每层要存入的个数

第四步:核心代码:

import math
n=int(input())
#print(int(math.log2(n)))
nums=list(map(int,input().split()))
arr=[[]for i in range(16)]
def twotree(n):
    index=0
    if n==1:
         return 1
    for i in range(1,int(math.log2(n))+2):
        #如何提取数组里的元素？index,index+pow ) 提取的数组索引
        for j in range(index,index+pow(2,i-1)):  #遍历几次就加几个数   
        # print(j) 
         if j<n:
            arr[i-1].append(nums[j])  
             #如何更新index的状态
        index=index+pow(2,i-1)
        if j>=n:
            break
        print(index)
    return arr 
print(twotree(n))
 

题目大部分问题解决了，现在解决最后一个问题，题目问的是权值最大在第几层，那这要先吧二维数组的索引求和，，
这里无意间找到一篇很好的案例，python的推导式格式固定写法，可惜没机会了，，，明天比赛了。
尝试了用numpy库，发现不是很理想，后来琢磨了用temp替换也行，

for index in range(len(arr)):
    temp=0
    for value in arr[index]:
        temp+=value
    arr[index]=temp
print(arr)
print(arr.index(max(arr)))

完整源码：

import math
n=int(input())
#print(int(math.log2(n)))
nums=list(map(int,input().split()))
arr=[[]for i in range(16)]
def twotree(n):
    index=0
    if n==1:
         return 1
    for i in range(1,int(math.log2(n))+2):
        #如何提取数组里的元素？index,index+pow ) 提取的数组索引
        for j in range(index,index+pow(2,i-1)):  #遍历几次就加几个数   
        # print(j) 
         if j<n:
            arr[i-1].append(nums[j])  
             #如何更新index的状态
        index=index+pow(2,i-1)
        if j>=n:
            break
        print(index)
    return arr 
print(twotree(n))
for index in range(len(arr)):
    temp=0
    for value in arr[index]:
        temp+=value
    arr[index]=temp
print(arr)
print(arr.index(max(arr)))

写了一个半小时，算是考前练手吧，希望明天不要太惨