[BAEKJOON] [10828] 스택: PYTHON

[10828] 스택

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성

10828번: 스택

Input

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

Output

• 출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력

 

✨ Solve

• 구현
  • 배열 혹은 연결리스트를 통해 구현 할 수 있음
  • 배열(리스트)을 통해 구현하는 방법이 더 쉬움

• 배열을 통한 스택 구현
  • 숫자를 담을 큰 배열 한개, 인덱스를 담을 변수 한개 필요
  • 변수는 다음에 삽입되어야 할 숫자가 들어가야할 인덱스 위치를 가르키고 있음
  • 원소는 0번지부터 담기므로, 변수는 스택 내에 담김 수의 개수를 나타내기도 함

 

💻 Code

### 언어 python3, 메모리 31256KB, 시간 56ms

stack = [0 for i in range(10000)]
pos = 0

def push(x: int):
    global pos
    stack[pos] = x
    pos +=  1
    
def pop():
    global pos
    if pos <= 0:
        return -1
    
    pos -= 1 
    # 들어있는 값 자체는 변경하지 않아도됨
    # 새로 들어오는 값이 덮어 쓸 예정

    return stack[pos]

def top():
    global pos
    if pos <= 0:
        return -1
    return stack[pos-1]

def size():
    global pos
    return pos

def empty():
    global pos
    if pos <= 0:
        return 1
    else:
        return 0


import sys

n = int(input())
while n > 0:
    line = sys.stdin.readline().rstrip().split()
    command = line[0]
    if command == 'push':
        push(line[1])
    elif command == 'top':
        print(top())
    elif command == 'pop':
        print(pop()) 
    elif command == 'size':
        print(size()) 
    elif command == 'empty':
        print(empty())
        
    n -= 1

 

🎯 Question

1. input()과  sys.stdin.readline()은 실행시간이 얼마나 차이날까?
    - 시간초과 일때  sys.stdin.readline() 쓰면 통과됨
2. 왜 list는 int 변수와 달리 global 을 안해줘도 적용이 되는걸까?