가오리의 코딩일기

두 배열의 원소 교체 본문

Python/이코테

두 배열의 원소 교체

류경혜 2022. 6. 29. 14:00

동빈이는 두 개의 배열 A와 B를 가지고 있다.

두 배열은 N개의 원소로 구성되어 있으며 배열의 원소는 모두 자연수이다.

동빈이는 최대 K번의 바꿔치기 연산을 수행할 수 있는데 바꿔치기 연산이란 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 것을 말한다.

동빈이의 최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록 하는 것이며 여러분은 동빈이를 도와야 한다

N,K 그리고 배열 A와 B의 정보가 주어졌을 때 최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력하는 프로그램을 작성하시오.

예를 들어 N=5, K=3이고 배열 A와 B가 다음과 같다고 하자

 

 - 배열 A = [1,2,5,4,3]

 - 배열 B = [5,5,6,6,5]

 

이 경우 다음과 같이 세 번의 연산을 수행할 수 있다.

 

 - 연산1) 배열 A의 원소 '1'과 배열 B의 원소 '6'을 바꾸기

 - 연산2) 배열 A의 원소 '2'과 배열 B의 원소 '6'을 바꾸기

 - 연산3) 배열 A의 원소 '3'과 배열 B의 원소 '5'을 바꾸기

 

세 번의 연산 이후 배열 A와 배열 B의 상태는 다음과 같이 구성될 것이다.

 

 - 배열 A = [6,6,5,4,3]

 - 배열 B = [3,5,1,2,5]

 

이때 배열 A의 모든 원소의 합은 26이 되며 이보다 더 합을 크게 만들 수는 없다. 따라서 이 예시의 정답은 26이 된다

 

입력조건

 - 첫 번째 줄에 N,K가 공백으로 구분되어 입력된다

 - 두 번째 줄에 배열 A의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10,000,000보다 작은 자연수이다.

 - 세 번째 줄에 배열 B의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10,000,000보다 작은 자연수이다.

 

출력조건

 - 최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력한다.

n, k = map(int,input().split())
aList = list(map(int,input().split()))
bList = list(map(int,input().split()))
aList = sorted(aList)
bList = sorted(bList)[::-1]
# aList.sort()
# bList.sort(reversed = True)
for i in range(k):
    if aList[i] <bList[i]:
        aList[i], bList[i] = bList[i], aList[i]
    else:
        break
print(sum(aList))

'Python > 이코테' 카테고리의 다른 글

떡볶이 떡 만들기  (0) 2022.07.05
부품 찾기  (0) 2022.07.04
성적이 낮은 순서로 학생 출력하기  (0) 2022.06.28
위에서 아래로  (0) 2022.06.27
미로 탈출  (0) 2022.06.26