[프로그래머스/Lv. 0] 다항식 더하기

import java.util.*;

class Solution {
    public String solution(String polynomial) {
        String answer = "";
        
        String[] pol = polynomial.split(" +");
        int linear = 0 ;
        int constant = 0;
        
        for(String s : pol) {
            if(s.equals("x")) {
                linear += 1;
            } else if(s.contains("x")) {
                linear += Integer.parseInt(s.substring(0, s.length() -1));
            } else if(!s.equals("+")) {
                constant += Integer.parseInt(s);
            }
        }
        
        // 일차항만 존재
        if(linear != 0 && constant == 0) {
            if(linear == 1) {
                answer += "x";
            } else {
                answer += linear + "x";
            }
        }
        
        // 둘다 존재
        if(linear != 0 && constant != 0) {
            if(linear == 1) {
                answer += "x" + " + " + constant;
            } else {
                answer += linear + "x" + " + " + constant;
            }
        }
        
        // 상수항만 존재
        if(linear == 0 && constant != 0) {
            answer += constant;
        }

        return answer;
    }
}

내가 놓친 점은 이제 if문 작성할 때도

x항만 존재하는 경우, 상수항만 존재하는 경우, 둘 다 존재하는 경우

이렇게 경우를 나눠줘서 생각해야했던..

 

---

(참고)

[ 자바 코딩 ] java substring 으로 문자열 자르기

 

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

[프로그래머스/Java] Lv.0 다항식 더하기