문제
옛날 옛적에 수학이 항상 큰 골칫거리였던 나라가 있었다. 이 나라의 국왕 김지민은 다음과 같은 문제를 내고 큰 상금을 걸었다.
길이가 N인 정수 배열 A와 B가 있다. 다음과 같이 함수 S를 정의하자.
S = A[0] × B[0] + ... + A[N-1] × B[N-1]
S의 값을 가장 작게 만들기 위해 A의 수를 재배열하자. 단, B에 있는 수는 재배열하면 안 된다.
S의 최솟값을 출력하는 프로그램을 작성하시오.
해결 방법
a리스트의 최솟값 * b리스트의 최댓값을 반복하면 해결된다! 그러면 각각 오름차순, 내림차순 정렬 후 풀면 간단하겠지만 b는 재배열하면 안 된다는 조건이..!있다..!
💡그러면 정렬이 아닌 min(), max() 함수로 각각의 최소, 최댓값을 뽑고 pop연산을 한다면 되겠군
위 과정을 n번 반복해주면 풀리는 문제이다.
제출
n = int(input())
a_list = list(map(int, input().split()))
b_list = list(map(int, input().split()))
ans = 0
cnt = 0
while True:
if cnt == n:
break
ans += min(a_list) * max(b_list)
a_list.remove(min(a_list))
b_list.remove(max(b_list))
cnt += 1
print(ans)메모
위처럼 while문 속 break 제어문을 사용하지 않고 for문으로 n번 반복해도 된다. 근데 왜 요즘 while문이 자꾸 쓰고싶지 저런 모양 코드가 좋니..? (. ❛ ᴗ ❛.)
