[백준] 1026번 보물(with python)

https://www.acmicpc.net/problem/1026

1026번: 보물

 

1. 내가 작성한 코드

n = int(input())
a = sorted(list(map(int, input().split())),reverse=True)
b = sorted(list(map(int, input().split())))

for i in a:
  if i == 0:
    a.remove(0)
    a.append(0)

print(sum(list(map(lambda x,y : x * y, a,b))))

아까 공부한 lambda를 이용해서 풀어봤는데, 조건을 다시 살펴보니 b는 재정렬하면 안된다.

 

따라서 lambda는 지우고, index, min, max 함수를 이욯애 풀었다.

 

2. 수정한 코드

n = int(input())
a = sorted(list(map(int, input().split())))
b = list(map(int, input().split()))
s = 0

for i in range(n):
  s += min(a) * max(b)
  a.pop(a.index(min(a)))
  b.pop(b.index(max(b)))

print(s)