프로그래머스/Lv. 1

[프로그래머스 코딩테스트] 두 정수 사이의 합(Java)

Sigfriede 2023. 4. 4. 01:36

  문제 설명

  두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.

  예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

 

  제한사항

  • a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
  • a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
  • a와 b의 대소관계는 정해져있지 않습니다.

 

  입출력 예

a b return
3 5 12
3 3 3
5 3 12
class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        long max = Math.max(a, b);
        long min = Math.min(a, b);
        if (a == b) {
            answer = a;
        } else {
            answer = (max - min + 1) * (max + min) / 2;
        }
        return answer;
    }
}

  max와 min은 각각 a와 b를 비교하여 큰 수와 작은 수를 할당하는 변수입니다. Math.max와 Math.min는 입력받은 숫자(인자) 중 최대값과 최소값을 수를 반환하는 메소드입니다. ab의 대소관계가 정해져 있지 않기 때문에 크기비교를 위해 사용했습니다. a와 b의 뺄셈이 있기 때문에, 이를 고려하지 않으면 a와 b의 크기를 비교하는 식을 세우거나 -를 처리하는 별도의 식이 필요합니다.

  a와 b가 같을 경우 아무 수나 리턴하라고 했으므로 a를 리턴하도록 했습니다. 그렇지 않을 경우, (b - a + 1) * (a + b) / 2로 등차수열의 합 공식을 사용했습니다.