华为0830-题解-频率搬移值分配| 二叉树构建+层序遍历

python

原题题目自行搜索

题解

思路:

dfs二叉树构建+层序遍历二叉树

  • dfs深度搜优先递归每个节点的值,同时根据dfs带来的前序遍历效果==构建二叉树==
  • 二叉树层序遍历
n = int(input())
leaves = list(map(int,input().split()))

class Tree():
    def __init__(self,val):
        self.left = None
        self.right = None
        self.val = val

c= []#测试使用,用于打印
#深度优先搜索的过程,顺便构造二叉树
def dfs(left,right,sumroot):
  
    if left>right:
        return 
    mid = (left+right)//2
    avg = (min(leaves[left:right+1])+max(leaves[left:right+1]))//2
    val = avg-sumroot
    c.append(val)
    node = Tree(val)
    if left==right:
        return Tree(val)

    node.left = dfs(left,mid,sumroot+val)
    node.right = dfs(mid+1,right,sumroot+val)
    return node

# root即为构造好的二叉树
root = dfs(0,len(leaves)-1,0)

#常规的二叉树层序遍历
from collections import deque
stack = deque()
stack.append(root)
res = []
while stack:
    node = stack.popleft()
    res.append(node.val)
    if node.left:
        stack.append(node.left)
    if node.right:
        stack.append(node.right)
#result
print(' '.join([str(item) for item in res]))





#offer等你##华为求职进展汇总##我的求职思考#
全部评论

相关推荐

不愿透露姓名的神秘牛友
07-16 14:00
白火同学:其实你可以了解一下HR在Boss聊天的机制,想赢牌的前提是先会玩牌。 如果HR长时间没有理你,有可能是因为你的消息被其他应聘者的消息给挤到下面了,HR从上到下有可能只看个三四百个人就要到理想数量的简历了,而你恰好没有被看到,时间一长,你的消息在越来越下面。这种情况就需要你自己活跃一下,把消息提上去。 也可能是HR招的合适的人选了,但会一直挂着岗位,为了省重新开招聘岗位的钱,方便后面随时修改招聘要求。 当然也可能是HR吃饱了没事耍你玩,要了你的简历又不看,就看你自己怎么理解了。
点赞 评论 收藏
分享
评论
2
2
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务