다항식 더하기

문제

---

나의 생각

다항식에 x값과 정수 값을 나누어 더하기를 시행만 해주면 된다. 

그렇다면 입력되는 매개변수를  " + ' 기준으로 나누어 배열에 넣고 남은 값들을 기준으로 x 가 있으면 계산을 별도로 진행, 정수는 바로 더하기 해주는 방식으로 접근. 

x 만 입력된 1x 가 될 수 있으니 해당 경우 return시 x로 변환하여 반환토록 진행.

---

나의 답안

public static String solution(String polynomial) {
    int sumA = 0;
    int sumB = 0;

    String[] split = polynomial.split(" [+] ");

    for (int i = 0; i < split.length; i++) {
        if (split[i].contains("x")) {
            if (split[i].length() == 1) {
                sumA += 1;
                continue;
            }
            String x = split[i].replace("x", "");
            System.out.println("x = " + x);
            sumA += Integer.parseInt(x);
        } else {
            sumB += Integer.parseInt(split[i]);
        }
    }
    if (sumA != 0 && sumB != 0) {
        if (sumA == 1){
            return "x + " + sumB;
        }
        return sumA + "x + " + sumB;
    } else if (sumA == 0) {
        return sumB + "";
    } else {
        if (sumA == 1){
            return "x";
        }
        return sumA + "x";
    }
}