华为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等你##华为求职进展汇总##我的求职思考#
全部评论

相关推荐

沉淀一会:1.同学你面试评价不错,概率很大,请耐心等待; 2.你的排名比较靠前,不要担心,耐心等待; 3.问题不大,正在审批,不要着急签其他公司,等等我们! 4.预计9月中下旬,安心过节; 5.下周会有结果,请耐心等待下; 6.可能国庆节前后,一有结果我马上通知你; 7.预计10月中旬,再坚持一下; 8.正在走流程,就这两天了; 9.同学,结果我也不知道,你如果查到了也告诉我一声; 10.同学你出线不明朗,建议签其他公司保底! 11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
2 2 评论
分享
牛客网
牛客企业服务