[백준] 10828번 스택

사용 언어 - Python3

10828번: 스택 (실버4, 스택)

문제 ★스택 문제★

10828번: 스택

 

 

정답

간단한 스택 문제 !!

이제 스택은 마스터한것 같아서 기분이 매우 좋네요 :)

(내 코드 풀이)

스택의 "선입후출" 구조만 이해하면 되는 문제

빈 리스트 stack을 선언한다.

리스트의 append(추가), pop(가장 마지막으로 들어온 원소 삭제)을 이용해서 각각의 명령문에 대한 코드를 작성한다.

# 정답

n = int(input())
command = [list(map(str,input().split())) for _ in range(n)]

stack = []

for i in range(n):
    
    if command[i][0] == 'push':
        stack.append(command[i][1])
    elif command[i][0] == 'pop':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop())
    elif command[i][0] == 'size':
        print(len(stack))
    elif command[i][0] == 'empty':
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    elif command[i][0] == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])

 

+ 나는 input을 사용해도 시간초과 에러가 안떴지만, 시간단축을 위해 sys.stdin.readline()을 이용하자!

+ for문 안에서 command를 입력받으면, 인덱싱도 command[0]으로 간단하게 할 수 있다.

for i in range(n):
    command = sys.stdin.readline().split()

    if command[0]=='push':
    	stack.append(command[1])

 

 

---

레퍼런스

• 정답 참고

[백준]스택/10828번/파이썬/Python/자료구조/스택

• 깃허브 정답

GitHub - yyeongeun/codingtest: 코딩테스트 공부