본문 바로가기
파이썬알고리즘

20210703#(58) 백준 10828번 스택

by zho 2021. 7. 3.

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

 

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

dfs,bfs 를 풀기전 먼저 기본적인 자료구조 개념들을 흡수하기 위해 풀어봤다!.. 첨엔 코드 이해하는게 꽤 시간이 걸렸는데 보다보니까 간단한 코드였다.

 

import sys 
N=int(sys.stdin.readline())
stack=[]

for _ in range(N):
    word=sys.stdin.readline().split() #input으로 받게 되면 시간 초과
    order=word[0] #예제에서 push 1, push 2 이런식으로 입력받게 됨. word=[push,1], word=[push,2] 이렇게 저장되는것임
    if order =="push":
        value=word[1] #위에서 받은것 중 숫자만 append하기 위한 추출
        stack.append(value)
    elif order =="pop": #pop()은 리스트의 맨 마지막 요소를 돌려주고 그 요소는 삭제함.
        if len(stack)==0:
            print(-1)
        else:
            print(stack.pop()) 
    elif order=="size":
        print(len(stack))
    elif order=="empty":
        if len(stack)==0:
            print(1)
        else:
            print(0)
    elif order=="top":
        if len(stack)==0:
            print(-1)
        else:
            print(stack[-1]) #맨 뒤에있는 값 print

 

다음은 큐다!! 화이팅

 

 

728x90