Study/Algorithm

프로그래머스-두 정수 사이의 합(java)

의미없는 시간은 없다. 2022. 7. 13. 02:48

문제 설명

두 정수 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의 대소관계는 정해져있지 않습니다.

--문제해결을 위해 먼저 대소관계가없는 두 숫자를 정렬하고 큰숫자에서 작은숫자를 뺀 숫자+1만큼 반복문을 돌면서 작은수에서 +1씩 증감을 한 값을 ArrayList에 넣고 ArrayList를 반복문을 돌려서 합을 구하였다.

import java.util.*;

class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        ArrayList<Integer> arr1 = new ArrayList<>();
        int temp;
        
        if(a==b){
            answer = a;
        }else{
            if(a>b){ //두개 수 정렬
                temp = b;
                b = a;
                a = temp;
            }
            temp = b-a;
            for(int i=0; i<temp+1; i++ ){
                arr1.add(a+i);
            }

            for(int j = 0 ; j<arr1.size();j++){
                answer+=arr1.get(j);
            }
        }
        return answer;
    }
}

--다른사람의 풀이

--역시 학생때 공부를 열심히 했어야했다. 등차수열을 까먹다니..

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;
    }
}

여기에서 Math.min, Math.max는 인자 두개가 같은 자료형이여야하는데 만약 다른 자료형이들어오면

ex) Math.min(3.14 , 2)를 하면, 2는 int형이므로 자동으로 형변환을하여 2.0을 반환한다.

반응형