파이썬) 백준_10828번

● 문제

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

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

 

● 입력

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

 

● 출력

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

 

● 코드

import sys

class Stack_1:
    def __init__(self):         #c=self가 됨
        self.a=[]      # 빈리스트에 선언
   
    def push(self, num):                           # Push 메소드 선언
        self.a.append(num)                         # 정수 추가        
           
    def pop(self):                                   
        if self.a == []:  
            print(-1)                            
            return                                
        else:                                        
            b=self.a[-1]                       # 맨 마지막 스택 출력
            self.a.pop(-1)                     # 스택에 가장 위에있는 정수 제거
            print(b)
            return                    
       
    def size(self):
        return print(len(self.a))                # 정수의 개수 출력
   
    def empty(self):                            
        if self.a ==[]:                          # 스택이 비면 1출력
            return print(1)
        else:
            return print(0)
   
    def top(self):
        if self.a==[]:                          # 스택이 비면 -1 출력
            print(-1)
            return
        else:                               # 가장 위에있는 정수 출력            
            print(self.a[-1])
            return
           
c=Stack_1()                                      # c라는 객체 선언          

print('명령의 수 : ',end='')
N=int(input())

for i in range(N):      # N개의 명령어 실행
    k=sys.stdin.readline().split()  
    d=k[0]
       
    if d == "push":
        c.push(k[1])
    elif d == "pop":    
        c.pop()
    elif d == "size":
        c.size()
    elif d == "empty":    
        c.empty()
    elif d == "top":
        c.top()

● 보완해야할 부분

    1. 객체 선언후 괄호 를 하자

    2. 인스턴스.메소드()  에서 괄호를 해주자! 

    3. 메소드 내부에 변수이름 설정 -> self.변수  로 해야한다.

 

 

 

 

출처) https://www.acmicpc.net/problem/10828