[BaekJoon] 1026번 : 보물 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 보물 입니다.

1026번: 보물

👨🏻‍💻 코드 ( Solution )

def treasure(A, B):
    answer = 0
    check_list = []
    
    for i in range(len(B)):
        check_list.append((i, B[i]))
    
    sorted_A = sorted(A)
    sorted_B = sorted(check_list, key=lambda x: -x[1])
    
    index_list = [idx[0] for idx in sorted_B]
    
    for i, idx in enumerate(index_list):
        answer += sorted_A[i] * B[idx]
        
    return answer

if __name__ == "__main__"   :
    N = int(input())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))
    
    print(treasure(A, B))

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07