Koala - 5기/코딩테스트 준비 스터디

[BOJ / Python] 1874번: 스택 수열

pcw 2022. 2. 11. 18:47

https://www.acmicpc.net/problem/1874

 

1874번: 스택 수열

1부터 n까지에 수에 대해 차례로 [push, push, push, push, pop, pop, push, push, pop, push, push, pop, pop, pop, pop, pop] 연산을 수행하면 수열 [4, 3, 6, 8, 7, 5, 2, 1]을 얻을 수 있다.

www.acmicpc.net

 

문제

 

풀이

첫째 줄에서는 n을 입력받고, 두번째 줄에서 네번째 줄까지는 arr이라는 배열을 생성하여 수열의 숫자를 입력해 저장하도록 한다. stack이라는 배열은 스택으로 사용하고 result라는 배열에는 +,-등의 값을 저장하도록 하며, idx 초기값은 0으로 설정한다. 반복문을 사용하여 숫자 i를 1부터 n까지 순차적으로 올리도록 하는데, append로 result와 stack에 각각 +와 숫자 i를 넣도록 한다. 여기서 arr배열의 idx위치에 있는 값과 i의 값이 같다면 result에 -를 넣고 pop를 사용하여 stack에서는 i값을 빼낸 뒤, idx의 값을 하나 증가시킨다. 그 뒤 while문을 사용하여 stack의 마지막 값과 arr배열의 idx위치에 있는 값이 다르다면 break문으로 while문을 빠져나오고, 같다면 위와 마찬가지로 result에 -를 넣고 pop로 stack에서의 i값을 빼낸 뒤 idx를 하나 증가시키는 방식을 반복한다. for 반복문이 모두 끝난 뒤 stack에 값이 남아있다면 NO를 출력시키고, 값이 남아있지 않다면 result값을 하나씩 출력시킨다. 이를 코드로 구현한 것은 아래와 같다.

 

코드

n=int(input())
arr=[]
for i in range(n):
  arr.append(int(input()))
stack=[]
result=[]
idx=0
for i in range(1,n+1):
  result.append("+")
  stack.append(i)
  if arr[idx]==i:
    result.append("-")
    stack.pop()
    idx+=1
  while stack:
    if stack[-1]!=arr[idx]:
      break
    else:
      result.append("-")
      stack.pop()
      idx+=1
if len(stack)!=0:
  print("NO")
else:
  for i in range(len(result)):
    print(result[i])

 

결과