[C++] SWEA 1224 - 계산기3

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV14tDX6AFgCFAYD&categoryId=AV14tDX6AFgCFAYD&categoryType=CODE&problemTitle=1224&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1 

SW Expert Academy

<풀이>

1. 계산식을 입력받는다.

2. 계산식을 후위표기식으로 바꾼 후에 계산하고 출력한다.

 

중위표기식을 후위표기식으로 바꾸는 방법과 후위표기식 계산방법은 설명하지 않겠습니다. 다른 자료를 참고해주세요. 정확한 구현을 요한 문제입니다.

 

#include <iostream>
#include <stack>
 
using namespace std;
 
int length;
string str;
int answer;
 
int priority(char c) {
    if (c == '(' || c == ')') return 0; //스택 안에 있는 괄호는 우선순위가 가장 낮음
    else if (c == '+' || c == '-') return 1;
    else if (c == '*' || c == '/') return 2;
}
 
//중위표기식 -> 후위표기식
string convert() {
    string output = "";
    stack<char> s;
    for (int i = 0; i < length; i++) {
        char c = str[i];
 
        //숫자
        if (0 <= 'c' && c >= '9') {
            output += c;
        }
        //그 외
        else {
            //괄호
            if (c == '(' || c == ')') {
                if (c == '(') s.push('(');
                else {
                    while (!s.empty() && s.top() != '(') {
                        output += s.top();
                        s.pop();
                    }
                    s.pop();
                }
            }
            //비괄호
            else {
                while (!s.empty() && priority(s.top()) >= priority(c)) {
                    output += s.top();
                    s.pop();
                }
                s.push(c);
            }
        }
    }
 
    while (!s.empty()) {
        output += s.top();
        s.pop();
    }
 
    return output;
}
 
//후위표기식 계산
int calculate(string str) {
    int output = 0;
    stack<int> num;
    for (int i = 0; i < str.length(); i++) {
        char c = str[i];
 
        if ('0' <= c && c <= '9') num.push(c - '0');
        else {
            int num1 = num.top();
            num.pop();
            int num2 = num.top();
            num.pop();
            switch (c) {
            case '+':
                num.push(num1 + num2);
                break;
            case '-':
                num.push(num1 - num2);
                break;
            case '*':
                num.push(num1 * num2);
                break;
            case '/':
                num.push(num1 / num2);
                break;
            }
        }
    }
 
    output = num.top();
    return output;
}
 
void solution() {
    string convertStr = convert();
    answer = calculate(convertStr);
}
 
int main() {
    int test_case;
    int T;
 
    T = 10;
 
    for (test_case = 1; test_case <= T; test_case++) {
 
        //초기화
        length = 0;
        str = "";
        answer = 0;
 
        //입력
        cin >> length >> str;
 
        //해법
        solution();
 
        //출력
        cout << "#" << test_case << " " << answer << "\n";
    }
 
    //종료
    return 0;
}

 

자료구조 스택과 구현에 대해 알아볼 수 있는 문제였습니다.