[프로그래머스] 다항식 더하기 (Java)

프로그래머스

접근 방식

문자열을 처리하여 x항의 계수와 상수 합의 결과값을 구하면 되는 비교적 간단한 문제이다. 더하기만 고려하면 되고 항의 순서가 뒤바뀌는 일도 없어 예외처리할 거의 없었다. 한 가지, x가 1일 경우 1x가 아닌 x로 반환해야한다는 점을 고려하지 못해 중간에 시간이 좀 걸렸다. 

로직 설명

- "+" 문자열을 기준으로 split()하여 strArr에 저장

- 계수와 상수를 저장할 int[] count 선언

- strArr에 저장된 각 항을 탐색하며:

1) x항인 경우, x를 없앤 나머지 숫자를 count[0]에 더하기2) 상수항인 경우, 숫자를 count[0]에 더하기- x항이 존재하는 경우 answer 문자열에 x항 표기 (1x인 경우 x로 표기)- x항이 존재하고 상수항이 있는 경우 answer 문자열에 "+" 문자열 표기- 상수항이 존재하는 경우 answer 문자열에 상수항 표기

class Solution {
    public String solution(String polynomial) {
        String answer = "";
        
        String[] strArr = polynomial.split(" \\+ ");
        
        int[] count = new int[2];
        
        for(String str : strArr){
            if(str.contains("x")){
                str = str.replace("x", "");
                count[0] += str.equals("") ? 1 : Integer.parseInt(str);
            }else{
                count[1] += Integer.parseInt(str);
            }
        }
        
        if(count[0] > 0){
            answer += count[0] == 1 ? "x" : count[0] + "x";
        }
        
        if(answer.length() > 0 && count[1] > 0){
            answer += " + ";
        }
        
        if(count[1] > 0){
            answer += count[1];
        }
        
        return answer;
    }
}

Note

- 새로 추가된 문제인데 난이도가 높은 문제들보다 점수를 많이 주길래 일단 남겨본다. 저런 예외들이 포함된 테스트케이스는 에러메시지가 없다면 혼자 모든 경우의 수를 상상해서 때려맞춰야 하기에 좋은 방식인지는 여전히 모르겠다.