배열 비교하기

문제 설명

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

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

두 정수 배열 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

---

import java.util.Arrays;

class Solution {
    public int solution(int[] arr1, int[] arr2) {
        int answer = 0;
       
        int arr1length = arr1.length;
        int arr2length = arr2.length;
        
        // 배열의 길이가 다르면 긴 쪽이 더 크다.
        if(arr1length > arr2length) {
            return answer = 1;
        } else if(arr1length == arr2length) {
            // 배열의 길이가 같으면 내부 원소를 모두 합한다.
            int arr1sum = Arrays.stream(arr1).sum();
            int arr2sum = Arrays.stream(arr2).sum();
            // 다를 때 왼쪽이 크면 1, 같으면 0, 오른쪽이 크면 -1을 리턴한다.
            // 같으면 같다.
            if(arr1sum > arr2sum) {
                return answer = 1;
            } else if(arr1sum == arr2sum) {
                return answer = 0;
            } else if(arr2sum > arr1sum) {
                return answer = -1;
            }
        } else if(arr2length > arr1length) {
            return answer = -1;
        }
        
        return answer;
    }
}

---

오늘은 우선 로직을 주석으로 먼저 적었다.

그러니까 내가 알아야 할 것이 보였다.

배열 안의 모든 원소를 어떻게 더할 것인가?

for문으로 다 돌리는 건 dumb하다고 생각했다.

그런데 stream으로 sum하는 문법이 기억이 안 나서

java array elements sum 키워드로 검색을 하니

https://www.geeksforgeeks.org/java-program-to-find-sum-of-array-elements/

Java Program to Find Sum of Array Elements - GeeksforGeeks

이 페이지가 나왔다.

다 읽어 봤는데 의외였던 건, for문으로 다 더하는 거나 stream으로 sum하는 거나

둘 다 시간 복잡도가 O(n)이라는 거다.

결국 '전부' 순회한다는 것...!

 

그래서 위 스타일대로 코드 작성을 끝냈다.

 

프로그래머스에서 가장 추천을 많이 받은 답변을 보니까 stream으로만 끝냈던데

그렇게 하는 건 리턴 값이 -1, 0, 1 이런 식으로 확실하게 나온 다는 걸 '확신'할 때만 사용 가능한 것 같다.