Learntosurf / 11월 1주차 / 2문제

[learntosurf]

주 목표 문제 수: 2개

푼 문제

• 백준 #28278. 스택2: 자료구조 / 실버4
• 백준 #18258. 큐2: 자료구조 / 실버4

---

🚩제출한 코드

import sys
input = sys.stdin.read
output = sys.stdout.write

def stack_operations(): 
    data = input().splitlines()  
    N = int(data[0])
    
    stack = []  
    results = [] 
    
    for i in range(1, N+1):
        command = data[i].split()
        
        if command[0] == '1':
            X = int(command[1])
            stack.append(X)
        elif command[0] == '2':
            if stack:
                results.append(str(stack.pop()))
            else:
                results.append('-1')
        elif command[0] == '3':
            results.append(str(len(stack)))
        elif command[0] == '4':
            results.append('1' if not stack else '0')
        elif command[0] == '5':
            if stack:
                results.append(str(stack[-1]))
            else:
                results.append('-1')
    
    output("\n".join(results) + "\n")

stack_operations()

import sys
from collections import deque
input = sys.stdin.read
output = sys.stdout.write

def queue_operations(N, data):
    queue = deque()
    results = []
    
    for i in range(1, N+1):
        command = data[i].split()
        
        if command[0] == 'push':
            X = int(command[1])
            queue.append(X)
        elif command[0] == 'pop':
            if queue:
                results.append(str(queue.popleft()))
            else:
                results.append('-1')
        elif command[0] == 'size':
            results.append(str(len(queue)))
        elif command[0] == 'empty':
            results.append('1' if not queue else '0')
        elif command[0] == 'front':
            if queue: 
                results.append(str(queue[0]))
            else:
                results.append('-1')
        elif command[0] == 'back':
            if queue:
                results.append(str(queue[-1]))
            else:
                results.append('-1')
        
    output("\n".join(results) + "\n")

data = input().splitlines()
N = int(data[0])
queue_operations(N, data)

[YoonYn9915]

👍