BOJ : 스택 수열 [1874]

재현·2021년 2월 6일

1874 algorithm boj python 백준 알고리즘 클론코딩 파이썬

분류 : 스택, 그리디

1. 문제

스택 (stack)은 기본적인 자료구조 중 하나로, 컴퓨터 프로그램을 작성할 때 자주 이용되는 개념이다. 스택은 자료를 넣는 (push) 입구와 자료를 뽑는 (pop) 입구가 같아 제일 나중에 들어간 자료가 제일 먼저 나오는 (LIFO, Last in First out) 특성을 가지고 있다.

1부터 n까지의 수를 스택에 넣었다가 뽑아 늘어놓음으로써, 하나의 수열을 만들 수 있다. 이때, 스택에 push하는 순서는 반드시 오름차순을 지키도록 한다고 하자. 임의의 수열이 주어졌을 때 스택을 이용해 그 수열을 만들 수 있는지 없는지, 있다면 어떤 순서로 push와 pop 연산을 수행해야 하는지를 알아낼 수 있다. 이를 계산하는 프로그램을 작성하라.

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

2. 아이디어

스택 마지막 수와 입력된 수열이 다르다면 NO

mine(clone1)

입력 루프, 숫자 스택과 결과 스택 생성

입력

입력을 받고 for문을 통해 리스트에 추가

숫자 스택

1부터 입력된 수열까지 (pop된 수 제외) 추가되는 스택

결과 스택(+,-)

숫자 스택 push : 결과 스택 + push

숫자 스택 pop : 결과 스택 - push

someone(clone2)

input() -> stdin.readline()

from sys import stdin

시간 감소 (3796ms -> 220ms)

mine(only)

2/11일 스스로 코딩할 예정

3. 코드

mine(clone)

n = int(input())
number = [] # 입력 원소 
for i in range(n):
  number.append(int(input()))

stack = [] #스택
result = [] #결과 +, -
maxNum = 1 # 현재 1부터 입력된 수열까지의 값 중 최댓값
for i in number:
  while maxNum <= i: # 1부터 입력된 수열까지
    stack.append(maxNum) # 스택에 1~i까지 push
    result.append('+') # 결과에 '+' push
    maxNum += 1
  if stack.pop() == i: # 스택 마지막 수와 입력된 수열이 같다면 pop
    result.append('-') # 결과에 '-' push
  else: # 스택 마지막 수와 입력된 수열이 다르다면 스택을 비운 후 'NO'만 push
    result.clear()
    result.append('NO')
    break # 이후 진행하지 않고 루프 탈출

for i in result: # 결과 출력
  print(i)

mine(only)

'''

출처 : https://github.com/Gitgorithm/wogus0333_Github/blob/main/BOJ/BOJ_1874.py

someone(clone2)

from sys import stdin

n = int(stdin.readline())
number = [] # 입력 원소 
for i in range(n):
  number.append(int(stdin.readline()))

stack = [] #스택
result = [] #결과 +, -
maxNum = 1 # 현재 1부터 입력된 수열까지의 값 중 최댓값
for i in number:
  while maxNum <= i: # 1부터 입력된 수열까지
    stack.append(maxNum) # 스택에 1~i까지 push
    result.append('+') # 결과에 '+' push
    maxNum += 1
  if stack.pop() == i: # 스택 마지막 수와 입력된 수열이 같다면 pop
    result.append('-') # 결과에 '-' push
  else: # 스택 마지막 수와 입력된 수열이 다르다면 스택을 비운 후 'NO'만 push
    result.clear()
    result.append('NO')
    break # 이후 진행하지 않고 루프 탈출

for i in result: # 결과 출력
  print(i)