[정렬 알고리즘] 두 배열의 원소 교체 - 파이썬(python)

두 배열의 원소 교체

난이도 : 下 풀이 시간 : 20분

시간 제한 : 2초 메모리 제한 : 128 MB

 

---

 

해답

 

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

a = list(map(int, input().split()))
b = list(map(int, input().split()))

a.sort()
b.sort(reverse=True)

for i in range(k):
    if a[i] > b[i]:
        a[i], b[i] = b[i], a[i]
    else:
        break

print(sum(a))

---

예시

 

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의 모든 원소의 합을 출력

---

해설

 

문제 해결 아이디어는 배열 A의 가장 작은 원소를 배열 B의 가장 큰 원소와 교체를 하는 것입니다.
하지만 배열 A의 최솟값이 배열 B의 최댓값보다 작을 때에만 교체를 수행해야 합니다.
배열 A를 오름차순으로, 배열 B를 내림차순으로 정렬한 후 첫 번째 원소부터 차례대로 비교하면 됩니다.
문제에서는 두 배열의 원소가 최대 100,000개까지 입력될 수 있으므로 O(NlogN)을 보장하는 정렬 알고리즘을 이용합니다.