728x90
문제
괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다. 여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나태내야 한다.
입력
첫째 줄에 괄호 문자열이 한 줄에 주어진다. 하나의 괄호 문자열의 길이는 2 이상 50 이하이다.
출력
만일 입력 괄호 문자열이 올바른 괄호 문자열(VPS)이면 “YES”, 아니면 “NO”를 한 줄에 하나씩 차례대로 출력해야 한다.
예제 입력
(())())
예제 출력
NO
#접근을 잘 못한 코드
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
import sys
getInfo = sys.stdin.readline()
left_stack = list()
right_stack = list()
flag = True
for i in range(len(getInfo)):
if getInfo[0] == ')' or getInfo[-1] == '(':
flag = False
break
elif getInfo[i] == '(':
left_stack.append('(')
elif getInfo[i] == ')':
right_stack.append(')')
if len(left_stack) == len(right_stack) and flag == True:
print('YES')
else:
print('NO')
|
cs |
#개선된 코드
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
import sys
getInfo = sys.stdin.readline()
score = 0
for i in range(len(getInfo)):
if getInfo[i] == '(':
score += 1
elif getInfo[i] == ')':
score -= 1
if score < 0:
print('NO')
break
if score > 0:
print('NO')
elif score == 0:
print('YES')
|
cs |
반응형