[Programmers] Lv.0 | 다항식 더하기 | Java

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120863

 

풀이

class Solution {
    public String solution(String polynomial) {
        int x = 0, n = 0;
        
        for (String term : polynomial.split(" ")) {
            if (term.equals("+")) {
                continue;
            }
            
            if (term.contains("x")) {
                x += term.equals("x") ? 1 : Integer.parseInt(term.replace("x", ""));
            } else {
                n += Integer.parseInt(term);
            }
        }
        
        // x항이 없으면 상수만 출력
        if (x == 0) {
            return String.valueOf(n);
        }
        // 상수항이 없으면 x항만 출력
        if (n == 0) {
            return x == 1 ? "x" : x + "x";
        }
        // x항과 상수항 모두 있을 때
        return (x == 1 ? "x" : x + "x") + " + " + n;
    }
}