m2skills

balanced brackets python

Apr 18th, 2017
283
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.21 KB | None | 0 0
  1. # Program to check balanced brackets using stack
  2. test = int(input())
  3. flag = False
  4. for k in range(test):
  5.     stack = []
  6.     flag = False
  7.     string = input()
  8.     for i in string:
  9.         if i == '{' or i=='(' or i=='[':
  10.             stack.append(i)
  11.         elif i == ')':
  12.             if len(stack)>0:
  13.                 element = stack.pop()
  14.                 if element != '(':
  15.                     flag = True
  16.                     break
  17.             else:
  18.                 flag = True
  19.                 break
  20.  
  21.         elif i == ']':
  22.             if len(stack) > 0:
  23.                 element = stack.pop()
  24.                 if element != '[':
  25.                     flag = True
  26.                     break
  27.             else:
  28.                 flag = True
  29.                 break
  30.  
  31.         elif i == '}':
  32.             if len(stack) > 0:
  33.                 element = stack.pop()
  34.                 if element != '{':
  35.                     flag = True
  36.                     break
  37.             else:
  38.                 flag = True
  39.                 break
  40.  
  41.     if not flag and len(stack) == 0:
  42.         print("YES")
  43.     else:
  44.         print("NO")
  45.        
  46. '''
  47.  
  48. Sample Input
  49.  
  50. 3
  51. {[()]}
  52. {[(])}
  53. {{[[(())]]}}
  54.  
  55.  
  56. Sample Output
  57.  
  58. YES
  59. NO
  60. YES
  61. '''
Add Comment
Please, Sign In to add comment