[Programmers] 두 정수 사이의 합

programmers.co.kr/learn/courses/30/lessons/12912

코딩테스트 연습 - 두 정수 사이의 합

 

Java

 

처음에 쓴 코드.. if ~ else if를 없애고 싶었는데 어떻게 해야할지 생각이 잘 안 났다ㅠ

public class Solution {
	public long solution(int a, int b) {
	long answer = 0;
		if(a == b) return a;
		else if(a > b) {
			for(int i = b; i <= a; ++i) {
				answer += i;
			}
		} else {
			for(int i = a; i <= b; ++i) {
				answer += i;
			}
		}
		return answer;
    }
}

다른 분 코드 보고 수정한 코드

삼항연산자를 써서 if~ else if를 없앴다.

public class Solution {
	public long solution(int a, int b) {
		long answer = 0;
		for(int i = ((a > b) ? b : a); i <= ((a > b) ? a : b); ++i) {
			answer += i;
		}
		return answer;
	}
}