2.[1주차] baekjoon 10773번 파이썬 문제풀이

백준 10773번 문제를 파이썬으로 풀어보겠다.

10773번 문제

10773번 예제

스택은 데이터의 삽입과 삭제가 데이터의 가장 한쪽 끝에서만 일어나는 자료구조로 가장 마지막에 삽입된 데이터가 가장 먼저 사용되거나 삭제된다.

데이터를 삽입하는 과정을 push, 가장마지막에 삽입한 데이터를 삭제하는 과정을 pop라고 부른다.

count = int(input()) # 입력받을 총 숫자의 수
stk = [] # 스택 리스트

for i in range(count): 
    num = int(input())
    if(num == 0): #num이 0이면 pop
        stk.pop() #pop()함수는 맨 마지막 원소를 리턴하고 해당 원소는 삭제함.
    else:
        stk.append(num) #그게 아니라면 append = push
        #append() 안에 값을 입력하면 새로운 요소를 array 맨 끝에 객체로 추가
print(sum(stk))

코드에 대한 설명은 주석을 참고하자.

 

예제에 맞게 잘 출력되는지 확인 후 제출한다.

정상적으로 잘 풀었다.