[Python/파이썬] 프로그래머스 최솟값 만들기

프로그래머스

코드

def solution(A,B):
    answer = 0
    A = sorted(A)
    B = sorted(B, reverse=True)
    for i in range(len(A)):
        answer += A[i] * B[i]
    return answer

배열 A는 오름차순 정렬, 배열 B는 내림차순 정렬한 뒤 같은 인덱스끼리 곱하면 최솟값이 된다.

아주 간단한 문제인데 답을 모르겠어서 한참 헤맸는데 답 보니까 약간 힘 빠진다...그런데 문제 모르겠어서 찾아보다가 더 간단하게 푼 다른 코드에서 처음보는 함수를 배웠다.

def solution(A,B):
    return sum(a*b for a, b in zip(sorted(A), sorted(B, reverse = True)))

zip이라는 함수는 처음 보는데 이 함수는 파이썬의 내장함수로, iterable한 객체를 인자로 받아서 같은 인덱스의 값들을 튜플 형태로 묶어서 반환해주는 함수라고 한다. 알아두면 유용하게 쓸 수 있을 것 같다!