honey_pot
[정렬] 6 - 4 두 배열의 원소 교체 본문
A와 B 두 배열은 N개의 원소로 구성되어 있으며, 배열의 원소는 모두 자연수이다. 이때 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 바꿔치기 연산을 수행할 수 있으며, 최대 K번의 바꿔치기 연산을 수행할 수 있다. 최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록하는 것이다.
입력 조건
- 첫 번째 줄에 N, K가 공백으로 구분되어 입력된다. ( 1<=N<= 100,000, 0<=K<=N)
- 두 번째 줄에 배열 A의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10,000,000보다 작은 자연수이다.
- 세 번째 줄에 배열 B의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10,000,000보다 작은 자연수이다.
출력 조건
- 최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 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)을 보장하는 정렬 알고리즘을 이용해야 한다.
'이것이 코딩테스트다' 카테고리의 다른 글
[이진 탐색] 7 - 2 부품 찾기 (0) | 2021.06.15 |
---|---|
[이진 탐색] 7 - 1 (0) | 2021.06.15 |
[정렬] 6 - 3 성적이 낮은 순서로 학생 출력하기 (0) | 2021.06.15 |
[정렬] 6 - 1 파이썬의 정렬 라이브러리 (0) | 2021.06.14 |
[정렬] 6 - 1 정렬 알고리즘 (0) | 2021.06.14 |
Comments