[알고리즘] 문자열 다루기 - 로마숫자(cpp)

로마인들은 로마 숫자를 사용했다. 1은 I, 5는 V, 10은 X로 표현했고, I, V, X가 나타내는 문자열은 각 글자가 나타내는 수의 합이다. 예를 들어 XVI = 10 + 5 + 1 = 16이다.
단, IV는 4이고 IX는 9이다. 예를 들어 IVI = 4 + 1 = 5이다. 
위의 예외에 추가로, IIV는 3이고 IIX는 8이라고 하자. 반드시 IIV, IIX는 3, 8이어야 한다.
즉 IIV는 1 + 1 + 5 = 7도 아니고, 1 + 4 = 5도 아니고, 반드시 3이어야 하는 것이다.

최대 길이가 10인 로마 숫자를 나타내는 문자열이 주어질 때 이 문자열이 나타내는 수를 출력하는 프로그램을 작성하시오.

입력
표준 입력으로 입력을 받는다. 최대 길이가 10인 문자열이 주어지고, 이 문자열은 I, V, X로만 이루어져 있다.

출력 
표준 출력으로 숫자 하나를 출력한다. 이 수는 입력된 문자열이 표현하는 수이다.

예제 입력 1
XIIV
예제 출력 1
13

---

 

문자열 다루기 과제이다!

IV는 4이고 IX는 9이다. & 반드시 IIV, IIX는 3, 8이어야 한다.

라는 조건이 있기 때문에, 이를 우선순위로 먼저 처리해주었다.

 

#include <iostream>
#include <vector>
#include <string>
using namespace std;

// I, II, IIV, IV, V, VI, VII, IIX, IX, X

int main(){
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);

    vector<string> RomaNums = {"IIV", "IV", "IIX", "IX"}; // 3, 4, 8, 9
    string numbers;
    int cnt = 0;

    cin >> numbers;

    for (int i = 0; i < 4; ++i) {
        int index = numbers.find(RomaNums[i]);
        if (index != string::npos) {
            switch(i){
                case 0:
                    numbers.erase(numbers.begin() + index, numbers.begin() + (index + 3));
                    cnt += 3;
                    break;
                case 1:
                    numbers.erase(numbers.begin() + index, numbers.begin() + (index + 2));
                    cnt += 4;
                    break;
                case 2:
                    numbers.erase(numbers.begin() + index, numbers.begin() + (index + 3));
                    cnt += 8;
                    break;
                case 3:
                    numbers.erase(numbers.begin() + index, numbers.begin() + (index + 2));
                    cnt += 9;
                    break;
            }
            i--;
        }
    }
    // cout << numbers << "\n";
    // cout << "count : " << cnt << "\n";
    int len = numbers.length();
    for (int i = 0; i < len; ++i)
    {
        if(numbers[i] == 'X') {
            numbers.erase(numbers.begin() + i);
            cnt += 10;
            i--;
        }
        if(numbers[i] == 'V') {
            numbers.erase(numbers.begin() + i);
            cnt += 5;
            i--;
        }
        if(numbers[i] == 'I') {
            numbers.erase(numbers.begin() + i);
            cnt += 1;
            i--;
        }
    }
    cout << cnt;
    cnt = 0;
    
    return 0;
}