swea 4874. Forth (후위 표기법 계산하기)

문제 😂

Forth라는 컴퓨터 언어는 스택 연산을 기반으로 하고 있어 후위 표기법을 사용한다. 예를 들어 3+4는 다음과 같이 표기한다.
 

3 4 + .
 

Forth에서는 동작은 다음과 같다.
 

숫자는 스택에 넣는다.

연산자를 만나면 스택의 숫자 두 개를 꺼내 더하고 결과를 다시 스택에 넣는다.

‘.’은 스택에서 숫자를 꺼내 출력한다.

 

Forth 코드의 연산 결과를 출력하는 프로그램을 만드시오. 만약 형식이 잘못되어 연산이 불가능한 경우 ‘error’를 출력한다.
 

다음은 Forth 연산의 예이다.
 

코드	출력
4 2 / .	2
4 3 - .	1

 

 

[입력]
 

첫 줄에 테스트 케이스 개수 T가 주어진다.  1≤T≤50
 

다음 줄부터 테스트 케이스의 별로 정수와 연산자가 256자 이내의 연산코드가 주어진다. 피연산자와 연산자는 여백으로 구분되어 있으며, 코드는 ‘.’로 끝난다.

나눗셈의 경우 항상 나누어 떨어진다.

 

[출력]
 

#과 1번부터인 테스트케이스 번호, 빈칸에 이어 계산결과를 정수로 출력하거나 또는 ‘error’를 출력한다.

 

내 풀🦷

# 4874. Forth

# 입력받기
T = int(input())
for tc in range(1, T + 1):
    # 여기서 문자로 받아서 나중에 숫자로 따로 바꿔줘야 함
    susik = input().split()
    # 마지막에 점이 있어서 미리 지워줌
    susik.pop()
    # 연산을 담아줄 리스트 만들기
    num = []
    # 결과값 1로 초기화, 0이면 에러로 출력, 1이면 리스트의 마지막 값 출력
    result = 1

    # 수식을 도는 동안
    for i in range(len(susik)):
        # 수식이 피연산자면
        if susik[i] not in ['+','-','*','/']:
            # 수식에 그 숫자를 정수로 바꿔서 넣어준다
            num.append(int(susik[i]))

        # 수식이 연산자면
        else:
            # 숫자가 두개 이상 남아있어야 연산자를 처리해 주는데 아니면
            # 맞는 수식이 아니니까 틀렸으므로 결과값을 0으로 바꿔준다
            if len(num) < 2:
                result = 0
                break

            else: # 리스트의 길이가 2 이상이면 맞는 수식이므로 진행
                n1 = num.pop() # 두가지 숫자를 뽑는거는 모든 과정에서 진행되므로 미리 뽑아줌
                n2 = num.pop()
                new = 0

                if susik[i] == '*':
                    new = n2 * n1
                    num.append(new) # 연산 후 다시 리스트에 넣어준다
                    # num.append(n2*n1) # => 이런식으로 바로 처리하는게 더 좋은 듯
                elif susik[i] == '+':
                    new = n2 + n1
                    num.append(new)
                elif susik[i] == '-':
                    new = n2 - n1
                    num.append(new)
                elif susik[i] == '/':
                    new = n2 // n1 
                    ####### 이떼 몫 연산자로 안해주면 나중에 .0이 나오는 케이스가 발생한다!!####
                    num.append(new)

    # 리스트의 길이가 1이 아니면 연산자와 숫자가 남아있거나
    # 아니면 아예 없어지는 틀린 수식이다
    if len(num) != 1:
            result = 0

    if result:
        print(f'#{tc} {num[-1]}')
    else:
        print(f'#{tc} error')

#####

이 문제 풀면서 2가지를 잘못했는데....

 

- 숫자를 정수로 안 받았으면서 나중에 int로 값을 넣지 않은 것..

- 나누기 연산자를 사용하면 딱 정수로 나눠 떨어지는 것이 아닌 경우도 발생하므로 // 몫 연산자를 사용해주어야 한다..

참고함면 좋을 개념들 ✍️

중위 표기법(infix notation)

: 연산자를 피연산자의 가운데 표기하는 방법

  ex) A + B (우리가 평소에 사용하는 방법)

 

후위 표기법(postfix notation)

: 연산자를 피연산자 뒤에 표기하는 방법

 ex) AB + (컴퓨터가 사용하는 방법)

 

후위 표기법의 수식을 스택을 이용해서 계산하기

1.  피연산자를 만나면 스택 push
2. 연산자를 만나면 필요한 만큼의 피연산자를 스택에서 pop 하여 연산하고, 연산결과를 다시 스택에 push
3. 수식이 끝나면, 마지막으로 스택을 pop하여 출력!!!