[프로그래머스] 배열 비교하기

문제 설명

 

이 문제에서 두 정수 배열의 대소관계를 다음과 같이 정의합니다.

- 두 배열의 길이가 다르다면, 배열의 길이가 긴 쪽이 더 큽니다.

- 배열의 길이가 같다면 각 배열에 있는 모든 원소의 합을 비교하여 다르다면 더 큰 쪽이 크고, 같다면 같습니다.

 

두 정수 배열 arr1과 arr2가 주어질 때, 위에서 정의한 배열의 대소관계에 대하여 arr2가 크다면 -1, arr1이 크다면 1, 두 배열이 같다면 0을 return 하는 solution 함수를 작성해 주세요.

 

 

제한사항

 

- 1 <= arr1의 길이 <= 100

- 1 <= arr2의 길이 <= 100

- 1 <= arr1의 원소 <= 100

- 1 <= arr2의 원소 <= 100

- 문제에서 정의한 배열의 대소관계가 일반적인 프로그래밍 언어에서 정의된 배열의 대소관계와 다를 수 있는 점에 유의해주세요.

 

 

입출력 예

 

arr1	arr2	result
[49, 13]	[70, 11, 2]	-1
[100, 17, 84, 1]	[55, 12, 65, 36]	1
[1, 2, 3, 4, 5]	[3, 3, 3, 3, 3]	0

 

 

입출력 예 설명

 

입출력 예 #1

- 예제 1번에서는 arr1의 길이는 2이고 arr2의 길이는 3으로 arr2의 길이가 더 깁니다. 따라서 arr2가 arr1보다 크므로 -1을 return 합니다.

 

입출력 예 #2

- 예제 2번에서는 arr1의 길이와 arr2의 길이가 4로 같습니다. arr1의 모든 원소의 합은 100 + 17 + 84 + 1 = 202이고 arr2의 모든 원소의 합은 55 + 12 + 65 + 36 = 168으로 arr1의 모든 원소의 합이 더 큽니다. 따라서 arr1이 arr2보다 크므로 1을 return 합니다.

 

입출력 예 #3

- 예제 3번에서는 arr1의 길이와 arr2의 길이가 5로 같고 각 배열의 모든 원소의 합 또한 15로 같습니다. 따라서 arr1과 arr2가 같으므로 0을 return 합니다.

 

 

 

 

 

위 문제를 맨 처음 접근할 때 크게 3가지의 경우로 나누어서 생각을 했다.

 

1. arr1의 길이가 arr2의 길이보다 클 때 (arr1의 길이 > arr2의 길이)

2. arr2의 길이가 arr1의 길이보다 클 때 (arr2의 길이 > arr1의 길이)

3. arr1의 길이와 arr2의 길이가 같을 때 (arr1의 길이 == arr2의 길이)

 

이렇게 3가지로 나누었다면 문제에 나와있는 것과 같이 arr1의 길이와 arr2의 길이가 같을 때, 각 배열에 있는 모든 원소의 합을 비교하여 다르다면 더 큰 쪽이 크게, 같다면 같게 작성해야한다.

 

그래서 아래와 같이 작성하였다.

class Solution {
    public int solution(int[] arr1, int[] arr2) {
        int answer = 0;

        // arr1의 길이 > arr2의 길이
        if (arr1.length > arr2.length) {
            answer = 1;
        }
        // arr1의 길이 < arr2의 길이
        else if (arr1.length < arr2.length) {
            answer = -1;
        }
        // arr1의 길이 == arr2의 길이
        else if (arr1.length == arr2.length) {
            int arr1sum = 0;
            int arr2sum = 0;

            for (int i = 0; i < arr1.length; i++) {
                arr1sum += arr1[i];
                arr2sum += arr2[i];
                // arr1의 합 == arr2의 합
                if (arr1sum == arr2sum) {
                    answer = 0;
                }
                // arr1의 합 > arr2의 합
                else if (arr1sum > arr2sum) {
                    answer = 1;
                }
                // arr1의 합 < arr2의 합
                else if (arr1sum < arr2sum) {
                    answer = -1;
                }
            }
        }

        return answer;
    }
}

위에서 언급한 것처럼 if-else문을 사용하여 크게 3가지 경우로 나누었고, 3가지 중 3번째 조건 같은 경우는 또 안에서 한 번 더 if-else문을 나누어서 작성하였다.

 

그러면 보다 더 효율적인 코드를 살펴보겠다.

import java.util.stream.IntStream;

class Solution {
    public int solution(int[] arr1, int[] arr2) {
        int answer = Integer.compare(arr1.length, arr2.length);

        if(answer == 0) {
            answer = Integer.compare(IntStream.of(arr1).sum(), IntStream.of(arr2).sum());
        }

        return answer;
    }
}

 

내가 작성한 것과 비교하면 매우 간단하다...

 

 

IntStream은 Java의 스트림 API 중 하나로, 주로 int 데이터의 스트림을 다룰 때 사용된다.

여기서는 배열의 합을 계산하기 위해 사용된다.

 

 

Integer.compare(x, y)는 두 정수를 비교하여 결과를 반환하는 메서드이다.

- x < y → -1 반환

- x == y → 0 반환

- x > y → 1 반환

여기서는 arr1과 arr2의 배열 길이를 비교한다.

- arr1.length가 arr2.length보다 크면 1 반환

- 같으면 0 반환

- 작으면 -1 반환

 

 

if (answer == 0)

- 배열의 길이가 같으면 answer는 0이 된다.

- 이 경우, 배열의 각 원소 합을 비교한다.

 

 

IntStream.of(arr1).sum()

- IntStream.of(arr1)는 배열 arr1을 스트림으로 변환한다.

- .sum()은 스트림의 모든 원소를 합산한다.

 

 

Integer.compare(IntStream.of(arr1).sum(), IntStream.of(arr2).sum())

- 배열 원소의 합을 비교한다.

- arr1의 합이 크면 1반환

- 같으면 0 반환

- 작으면 -1 반환

 

 

 

사실 이는 내가 자바의 내부 메서드가 정확히 무엇이 있는지 몰랐기 때문일 것이다. 앞으로 알고리즘을 더욱더 풀면서 유용한 내부 메서드를 공부해나가는 수 밖에 없다.