코딩테스트/Programmers

[Programmers] 두 정수 사이의 합

주니어주니 2023. 8. 2. 16:50

 

 

* 내 풀이 

쉬운 문제였는데 더위 먹었나 별 희한하게도 풀었네 

import java.util.*;

class Solution {
    public long solution(int a, int b) {
        long answer = 0;
        
        if (a == b) return a;
        
        int min = Math.min(a, b); 
        int max = Math.max(a, b); 
        int[] arr = new int[max - min + 1];
        
        for(int i=0; i<arr.length; i++) {
            arr[i] = min;
            min++;
        }
        return Arrays.stream(arr).sum();
    }
}

시간초과 미쳤음,,,

 

 

* 다른 사람 풀이 ㅠ

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