[프로그래머스 Lv. 1] 두 정수 사이의 합 - JAVA

문제

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

 

class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        return answer;
    }
}

 

제한사항

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

 

입출력 예

a	b	return
3	5	12
3	3	3
5	3	12

 

---

문제 풀이

분석

a와 b의 대소관계는 정해져있지 않습니다. 주의! 

- 처음엔 단순히 for문으로 했다가 대소관계 제한 조건을 발견했다.

 

 

내 풀이

 

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

 

Math 함수를 이용하는 방법

class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        if(a==b) return a;
        int start = Math.min(a,b);
        int end = Math.max(a,b);
        for(int i=start;i<=end;i++){
            answer += i;
        }
        return answer;
    }
}

 

다른 풀이 1

- 등차수열의 합 공식 이용

class Solution {

    public long solution(int a, int b) {
        return sumAtoB(Math.min(a, b), Math.max(b, a));
    }

    private long sumAtoB(long a, long b) {
        return (b - a + 1) * (a + b) / 2;
    }
}

 

추가로 공부할 부분

수학 공식을 활용하면 더 깔끔하게 풀 수 있는 것 같다

다들 어떻게 생각하는지 대단해..

 

---

Lv.1  / Java / 연습문제

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