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

2022. 2. 11. 18:47· Koala - 5기/코딩테스트 준비 스터디

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])

 

결과

저작자표시 (새창열림)

'Koala - 5기 > 코딩테스트 준비 스터디' 카테고리의 다른 글

[BOJ / Swift & Python] 13975 - 파일 합치기 3  (0) 2022.02.13
[BOJ / JAVA] 2164 - 카드2  (0) 2022.02.12
[BOJ/C++] 1202번: 보석도둑  (0) 2022.02.11
[BOJ / C++] 2164 : 카드2  (0) 2022.02.10
[BOJ/c++] 2110- 종이의 개수  (0) 2022.02.07
'Koala - 5기/코딩테스트 준비 스터디' 카테고리의 다른 글
  • [BOJ / Swift & Python] 13975 - 파일 합치기 3
  • [BOJ / JAVA] 2164 - 카드2
  • [BOJ/C++] 1202번: 보석도둑
  • [BOJ / C++] 2164 : 카드2
KauKoala
KauKoala
항공대 알고리즘 동아리 Koala 🥰
KauKoala
Koala
KauKoala
전체
오늘
어제
  • 분류 전체보기 (1884)
    • 공지 게시판 (10)
    • 정보 게시판 (8)
    • Codeforce (15)
    • acm-icpc (6)
    • Koala - 1기 (16)
    • Koala - 2기 (111)
      • Programming Contest (1)
      • A반 (20)
      • B반 (39)
      • C반 (22)
      • 기초 강의 (18)
    • Koala - 3기 (10)
      • 기초 스터디 (7)
    • Koala - 4기 (67)
    • Koala - 5기 (144)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (68)
    • Koala - 6기 (102)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (25)
      • 모의 테스트 스터디 (1)
    • Koala - 7기 (167)
      • 기초 알고리즘 스터디 (97)
      • 코딩테스트 준비 스터디 (68)
      • 모의 테스트 스터디 (1)
    • Koala - 8기 (44)
      • 기초 알고리즘 스터디 (32)
      • 코딩테스트 준비 스터디 (10)
      • 코드포스 버츄얼 스터디 (0)
      • 프로그래머스 LV2 스터디 (0)
    • Koala - 9기 (205)
      • 기초 알고리즘 스터디 (138)
      • 코딩테스트 준비 스터디 (64)
      • 모의테스트 준비 스터디 (1)
    • Koala - 10기 (117)
      • 기초 알고리즘 스터디 (30)
      • 코딩테스트 준비 스터디 (86)
      • 모의테스트 준비 스터디 (1)
    • Koala - 11기 (151)
      • 기초 알고리즘 스터디 (46)
      • 코딩테스트 준비 스터디 (104)
      • 모의테스트 준비 스터디 (1)
    • Koala - 12기 (86)
      • 기초 알고리즘 스터디 (31)
      • 코딩테스트 준비 스터디 (55)
    • Koala - 13기 (119)
      • 기초 알고리즘 스터디 (52)
      • 코딩테스트 준비 스터디 (67)
    • Koala - 14기 (116)
      • 기초 알고리즘 스터디 (39)
      • 코딩테스트 준비 스터디 (77)
    • Koala - 15기 (138)
      • 기초 알고리즘 스터디 (73)
      • 코딩테스트 준비 스터디 (65)
    • Koala - 16기 (47)
      • 코딩테스트 기초 스터디 (16)
      • 코딩테스트 심화 스터디 (31)
    • Koala - 17기 (62)
      • 코딩테스트 기초 스터디 (15)
      • 코딩테스트 심화 스터디 (47)
    • Koala - 18기 (31)
      • 코딩테스트 기초 스터디 (11)
      • 코딩테스트 심화 스터디 (20)
    • Koala - 19기 (38)
      • 코딩테스트 기초 스터디 (7)
      • 코딩테스트 심화 스터디 (31)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

  • 🐨항공대 알고리즘 학회 Koala 3기 모집
  • 🐨항공대 알고리즘 학회 Koala 2기 모집
  • 소모임 소개

인기 글

태그

  • 백준
  • 파이썬
  • BFS
  • dfs
  • C++
  • BOJ
  • 백트래킹
  • dp

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.3.0
KauKoala
[BOJ / Python] 1874번: 스택 수열
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.