LintCode 367 [Expression Tree Build]

原题

表达树是一个二叉树的结构,用于衡量特定的表达。所有表达树的叶子都有一个数字字符串值。而所有表达树的非叶子都有另一个操作字符串值。
给定一个表达数组,请构造该表达的表达树,并返回该表达树的根。

对于 (2*6-(23+7)/(1+2)) 的表达(可表示为 ["2" "*" "6" "-" "(" "23" "+" "7" ")" "/" "(" "1" "+" "2" ")"]). 其表达树如下:

                 [ - ]
              /          \
         [ * ]              [ / ]
       /     \           /         \
     [ 2 ]  [ 6 ]      [ + ]        [ + ]
                      /    \       /      \
                    [ 23 ][ 7 ] [ 1 ]   [ 2 ] 

在构造该表达树后,你只需返回根节点[-]。

解题思路

  • 首先定义一个MyTreeNode,对于每一个符号或者数字,附上一个val,以便于之后evaluate它,来建立树
  • 如果expression[i]等于"("或者")"调整base然后continue跳过后面的步骤
  • 维护一个递增stack,建立最小树,类似题[Max Tree]
  • 最后递归copy已经建立的树,给self.exp_node赋值

完整代码

"""
Definition of ExpressionTreeNode:
class ExpressionTreeNode:
    def __init__(self, symbol):
        self.symbol = symbol
        self.left, self.right = None, None
"""
class MyTreeNode:
    def __init__(self, val, s):
        self.left = None
        self.right = None
        self.val = val
        self.exp_node = ExpressionTreeNode(s)

class Solution:
    # @param expression: A string list
    # @return: The root of expression tree
    def get_val(self, a, base):
        if a == '+' or a == '-':
            if base == sys.maxint:
                return base
            return 1 + base
        if a == '*' or a == '/':
            if base == sys.maxint:
                return base
            return 2 + base
        return sys.maxint
    
    def create_tree(self, expression):
        stack = []
        base = 0
        for i in range(len(expression)):
            if expression[i] == '(':
                if base != sys.maxint:
                    base += 10
                continue
            elif expression[i] == ')':
                if base != sys.maxint:
                    base -= 10
                continue
            val = self.get_val(expression[i], base)
    
            node = MyTreeNode(val, expression[i])
            while stack and val <= stack[-1].val:
                node.left = stack.pop()
            if stack:
                stack[-1].right = node
            stack.append(node)
        if not stack:
            return None
        return stack[0]
    
    def copy_tree(self, root):
        if not root:
            return None
        root.exp_node.left = self.copy_tree(root.left)
        root.exp_node.right = self.copy_tree(root.right)
        return root.exp_node
    
    def build(self, expression):
        # write your code here
        root = self.create_tree(expression)
        return self.copy_tree(root)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 第5章 引用类型(返回首页) 本章内容 使用对象 创建并操作数组 理解基本的JavaScript类型 使用基本类型...
    大学一百阅读 3,270评论 0 4
  • importUIKit classViewController:UITabBarController{ enumD...
    明哥_Young阅读 3,880评论 1 10
  • 如今我所在的城市––南京,每年的12.13都会拉响防空警报,国家公祭日,提醒大家勿忘国耻。 而我去过的另外两个城市...
    Coach张小鹿阅读 827评论 0 0
  • 江畔 风, 汽笛拉响时间的诱惑 建筑的尖顶敲响此下的钟声 与她合照像是可以炫耀的骄傲 背倚栏杆看着密行的人群 听着...
    Cactussnow阅读 191评论 0 0
  • 2q全文595字,19幅图片,建议阅读2分钟。 十一黄金周,天安门、故宫、王府井、西单、南锣鼓巷、颐和园、雍和宫等...
    小七君阅读 506评论 0 6