[프로그래머스-코딩테스트 입문] 다항식 더하기

Hi, There!
안녕하세요, 바오밥입니다.

---

목차

• 문제
• 풀이

---

문제

문제 내용

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

프로그래머스

---

풀이

나의 풀이

- x만 입력된 경우, x가 1인 경우 등 경우의 수를 적절히 계산해 주지 않으면 반례 발생

class Solution {
    public String solution(String polynomial) {
        String[] split_polynomial = polynomial.split(" ");
        
        int x = 0;
        int n = 0;
        
        for(String s : split_polynomial) {
            if(s.contains("x")) {
                if(s.length() == 1) x++;
                else x += Integer.valueOf(s.substring(0, s.length()-1));
            }
            else if(!s.contains("+")) n += Integer.valueOf(s);
        }
        
        StringBuilder sb = new StringBuilder();
        if(x > 0 && n > 0 && x!=1) sb.append(x+"x + "+n);    
        else if(x > 0 && n > 0 && x==1) sb.append("x + "+n);
        else if(x == 1) sb.append("x");
        else if(x > 0) sb.append(x+"x");
        else if(n > 0) sb.append(n);
        
        return sb.toString();
    }
}