👨🏻‍💻Computer Science/💻Algorithm

[프로그래머스] 두 정수 사이의 합 - Java

급할수록도라에몽 2022. 12. 9. 00:41
728x90

프로그래머스 두 정수 사이의 합 - Java

Description

두 정수 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;
        for (int i = ((a < b) ? a:b); i <= ((a<b)?b:a); i++)
           answer += i;
         return answer;
   }
class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        
        if(a>b){
            for(;b<=a;b++){
                answer+=b;
            }
        }else{
            for(;a<=b;a++){
                answer +=a;
            }
        }   
        return answer;
    }
}

처음에 삼항 연산자를 이용해 풀고 if else문과 for 반복문으로도 문제를 풀어보았다.

자바 책에서 삼항 연산자가 더 효율적이라는 것을 본 뒤로 어떻게든 삼항연산자로 풀려는 강박같은게 생겨버렸다.

 

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;
    }
} //등차수열을 사용한 다른 사람 풀이

등차수열의 공식을 이용한 함수 sumAtoB를 만든다.

그리고 Math의 .min .max 으로 a b 를 구분한 뒤에 sumAtoB 사용해 푼 문제해결이다.

등차수열 합의 공식 n * (a + b) /2 를 사용해 푼 신박한 접근이라고 생각한다.

코드에 (b - a + 1) 이라고 되어있는 것은 최소값부터 최댓값까지의 n 을 의미한다.