Level 0. 다항식 더하기

Level 0. 다항식 더하기

 

 

istringstream을 이용해 파싱하여 해결했다.

 

특별히 더 고민해 볼 것은 없는 문제 같다.

 

#include <iostream>
#include <sstream>
#include <string>

using namespace std;

string makePoly(int x, int c)
{
    if (x == 0)
        return to_string(c);
    else
        return (x == 1 ? "" : to_string(x)) + "x" + (c == 0 ? "" : (" + " + to_string(c)));
}

string solution(string polynomial)
{
    istringstream iss(polynomial);
    string s;
    int x = 0, c = 0;

    while (iss >> s)
    {
        size_t length = s.length();

        if (s == "+")
            continue;
        else if (s[length - 1] == 'x')
            x += (length == 1) ? 1 : stoi(s.substr(0, length - 1));
        else
            c += stoi(s);
    }

    return makePoly(x, c);
}