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

문제 설명

두 정수 배열 arr1과 arr2가 주어질때,

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

 

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

코드 설명

arr의 길이를 비교

길이가 같지 않을때 arr2가 크면 1을 리턴, 그렇지 않을때 -1을 리턴

길이가 같을때 배열의 원소의 합을 비교해서 arr2가 크면 -1을 arr1이 크면 1을 리턴

코드

def solution(arr1, arr2):
    x, y = len(arr1), len(arr2)
    if x != y:
        return 1 if x > y else -1
    elif x == y:
        sum_x, sum_y = sum(arr1), sum(arr2)
        if sum_x > sum_y:
            return 1 
        elif sum_x < sum_y: 
            return -1 
        else:
            return 0