[정렬] -두 배열의 원소 교체

N개의 원소를 가진 서로다른 배열 A,B가 있을 때 K번 교환하여 A의 원소의 총합을 최대로 하라.

 

입력 예시)

5 3

1 2 5 4 3

5 5 6 6 5

 

출력 예시)

26

 

 

나의코드

n,k=map(int,input().split())


arrA=list(map(int,input().split()))
arrB=list(map(int,input().split()))
arrA.sort()
arrB.sort(reverse=True)

for i in range(k):
    arrA[i],arrB[i]=arrB[i],arrA[i]


print(sum(arrA))

각각의 배열을 입력 받은뒤 A배열은 오른차순으로 B배열은 내림차순으로 정렬한다.

그 후 k번 만큼 0번 인덱스 부터 스와프 해준다.

 

 

정답 코드

n, k = map(int, input().split()) # N과 K를 입력 받기
a = list(map(int, input().split())) # 배열 A의 모든 원소를 입력받기
b = list(map(int, input().split())) # 배열 B의 모든 원소를 입력받기

a.sort() # 배열 A는 오름차순 정렬 수행
b.sort(reverse=True) # 배열 B는 내림차순 정렬 수행

# 첫 번째 인덱스부터 확인하며, 두 배열의 원소를 최대 K번 비교
for i in range(k):
    # A의 원소가 B의 원소보다 작은 경우
    if a[i] < b[i]:
        # 두 원소를 교체
        a[i], b[i] = b[i], a[i]
    else: # A의 원소가 B의 원소보다 크거나 같을 때, 반복문을 탈출
        break

print(sum(a)) # 배열 A의 모든 원소의 합을 출력

 

아.. 반례를 생각 못했다..!

B의 배열의 가장 큰값이 A의 가장 작은 값보다 더 작을 수도 있는 걸 놓침 !