프로그래머스 다항식 더하기 C++

프로그래머스 : 다항식 더하기

프로그래머스

 

---

- 문제

  다항식이 주어지면 해당 다항식을 정리하는 문제다.

 

---

- 풀이

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

string solution(string polynomial) {
    string answer = "";
    stringstream ss(polynomial);
    int x = 0, dig = 0;
    vector<string> s = {};
    string temp;
    while(ss >> temp)
        s.push_back(temp);
    for(auto a : s){
        if(a.find('x') != string::npos){
            a.erase(a.begin() + a.find('x'));
            if(a.empty())
                x += 1;
            else
                x += stoi(a);
        }
        else if(isdigit(a[a.size()-1]) != 0){
            dig += stoi(a);
        }
    }
    if(x > 0){
        if(x == 1)
            answer += "x";
        else
            answer += (to_string(x) + "x");
        if(dig > 0){
            answer += (" + " + to_string(dig));
        }
    }
    else if( x == 0){
        if(dig > 0)
            answer += to_string(dig);
        // else
        //     answer += "0";
    }
    return answer;
}

 

 

1x일때 1x가 아니라 x로 출력해야하는 부분때문에 오래 헤맸었다.

 

x가 나오게 만들어야 한다는 걸 주의할 것!

 

  

---

- 기억할 것!

 X