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

 

 

 

 

문제

 

 

 

 

 

 

풀이

import java.util.ArrayList;
import java.util.List;

class Solution {
    public String solution(String polynomial) {
        String answer = "";
		String[] values = polynomial.split(" \\+ ");
		int x_value = 0;
		int int_value = 0;
		
		for(int i=0; i<values.length; i++) {
			if(values[i].contains("x")) {
				if(values[i].length()>1) {
					x_value += Integer.parseInt(values[i].substring(0, values[i].length()-1));
				}else {
					x_value ++;
				}
			}else {
				int_value += Integer.parseInt(values[i]) ;
			}
		}
		
		if(x_value != 0 && int_value == 0) {
			if(x_value == 1) {
				answer = "x";
			}else {
				answer = Integer.toString(x_value) + "x";
			}
		}
		if(x_value != 0 && int_value != 0) {
			if(x_value == 1) {
				answer = "x + " + Integer.toString(int_value);
			}else {
				answer = Integer.toString(x_value) + "x + " + Integer.toString(int_value);
			}
		}
		if(x_value == 0 && int_value != 0) {
			answer = Integer.toString(int_value);
		}
        
        return answer;
    }
}