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

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

프로그래머스

 

저는 구현을 통해 문제를 해결했습니다.

8, 10 ,12 번에서 틀리신다면 답을 출력할 때 x를 1x로 표시하고 있는지를 확인해보시면 좋을 것 같습니다.

 

import java.util.*;
class Solution {
    public String solution(String polynomial) {
        String answer = cal(polynomial);
        return answer;
    }
    private static String cal(String polynomial){
        String[] temp = polynomial.split(" ");
        int xCnt = 0;
        int num = 0;
        // System.out.println(Arrays.toString(temp));
        for(String str : temp){
            if(str.equals("+")){
                continue;
            }
            if(str.charAt(str.length()-1) == 'x'){
                if(str.length() == 1){
                    xCnt++;
                }else{
                    xCnt += Integer.parseInt(str.substring(0,str.length()-1));
                }
            }else{
                num += Integer.parseInt(str);
            }
        }
        StringBuilder sb = new StringBuilder();
        if(xCnt == 1){
            sb.append("x ");
        }else if(xCnt != 0){
            sb.append(xCnt+"x ");
        }
        if(num != 0){
            if(sb.length() != 0){
                sb.append("+ ");
            }
            sb.append(num);
        }else{
            sb.setLength(sb.length()-1);
        }
        return sb.toString();
    }
}