Python P1739 表达式括号匹配 (栈的使用)

题目描述

假设一个表达式有英文字母（小写）、运算符（+、-、*、/）和左右小（圆）括号构成，以 @ 作为表达式的结束符。请编写一个程序检查表达式中的左右圆括号是否匹配，若匹配，则输出 YES；否则输出 NO。表达式长度小于 255，左圆括号少于 20 个。

输入格式

一行：表达式。

输出格式

一行：YES 或 NO。

输入输出样例

输入 #1

2*(x+y)/(1-x)@

输出 #1

YES

输入 #2

(25+x)*(a*(a+b+b)@

输出 #2

NO

说明/提示

表达式长度小于 255，左圆括号少于 20 个。

方法一：

s = input()
flag = 0
for i in s:
    if i == '(' :
        if flag:
            flag = flag*10+1
        else:
            flag += 1
    if i == ')' :
        if flag == 0:
            flag = 1
            break
        if flag != 1:
            flag = (flag-1)/10
        else:
            flag -= 1
if flag == 0:
    print('YES')
else:
    print('NO')

方法二： （使用栈，逻辑上更加清晰）

class Stack:
    def __init__(self):
        self.items = []
        
    def isEmpty(self):
        return self.items == []
    
    def push(self, item):
        self.items.append(item)
        
    def pop(self):
        return self.items.pop()
    
    def peek(self):
        return self.items[len(self.items)-1]
    
    def size(self):
        return len(self.items)
    
def parChecker(symbolString):
    s = Stack()
    balanced = True
    index = 0
    while index < len(symbolString) and balanced:
        symbol = symbolString[index]
        if symbol == '(':
            s.push(symbol)
        elif symbol == ')':
            if s.isEmpty():
                balanced = False
            else:
                s.pop()
        index += 1
    if balanced and s.isEmpty():
        return 'YES'
    else:
        return 'NO'

x = input().strip()
print(parChecker(x))