Python 感谢C++大佬提供的思路

s = input()
flag = True
top1 = -1
top2 = -1

for ch in s:
    if ch == '(':
        top1 = top1 + 1
    
    elif ch == ')':
        top1 = top1 - 1
        if top1 < -1:
            flag = False
    
    elif ch == '[':
        top2 = top2 + 1
    
    elif ch == ']':
        top2 = top2 - 1
        if top2 < -1:
            flag = False

for i in range(0, len(s) - 1):
    if s[i:i + 2] == '(]' or s[i:i + 2] == '[)':
        flag = False

if top1 == -1 and top2 == -1 and flag:
    print("OK")
else:
    print("Wrong")