[programmers] 다항식 더하기

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

프로그래머스

 

---

< 다항식 더하기 >

 

문제 풀이 (Java)

class Solution {
	public String solution(String polynomial) {
		String answer = "";

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

		int x = 0, y = 0;
		for (int i = 0; i < arr.length; i += 2) {
			if (arr[i].charAt(arr[i].length() - 1) == 'x') {
				if (arr[i].length() == 1) {
					x += 1;
				} else {
					x += Integer.parseInt(arr[i].substring(0, arr[i].length() - 1));
				}
			} else {
				y += Integer.parseInt(arr[i]);
			}
		}

		if (x > 0) {
			if (x > 1) {
				answer += Integer.toString(x);
			}
			answer += "x";
		}
		if (y > 0) {
			if (answer.length() > 0) {
				answer += " + ";
			}
			answer += Integer.toString(y);
		}

		return answer;
	}
}

 

수식을 공백기준으로 나눈다. 덧셈을 제외하고 살펴보며 일차항이라면 일차항끼리 더하고, 상수항이라면 상수항끼리 더한다. 최종 answer에 일차항과 상수항을 저장한 후 반환한다. 

---

---

 

출처: 프로그래머스 코딩 테스트 연습, 
https://school.programmers.co.kr/learn/challenges