파이썬 알고리즘-106 (프로그래머스) 최솟값 만들기

7073 단어 algorithmalgorithm

코드

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

다른 사람의 풀이

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

zip()

인자로 iterbale 객체를 받아 같은 인덱스끼리 튜플로 모아서 돌려준다.
그냥 zip을 출력해 보면 zip 객체가 나오므로 값을 보려면 list 등으로 묶어야 한다.
인자가 하나만 있을 경우에는 각각의 원소를 튜플로 감싼 객체를 리턴하고, 길이가 다른 iterable이 여러 개를 인자로 받을 경우에는 가장 짧은 길이에 맞춘 객체를 리턴한다.

number_list = [1, 2, 3]
str_list = ['one', 'two', 'three']
result = zip(number_list, str_list)

print(list(result)

>> [(1, 'one'), (2, 'two'), (3, 'three')]
numbersList = [1, 2, 3]
str_list = ['one', 'two']
numbers_tuple = ('ONE', 'TWO', 'THREE', 'FOUR')
result = zip(numbersList, str_list, numbers_tuple)

print(list(result))

>> [(1, 'one', 'ONE'), (2, 'two', 'TWO')]

좋은 웹페이지 즐겨찾기