[PART2] 6-3(정렬): 두 배열의 원소 교체
💥이코테 실전문제 뽀개기💥
💻 6-3 두 배열의 원소 교체
난이도🖤🤍🤍 | 풀이시간 20분 | 제한시간 2초 | 메모리제한 128MB | 기출 국제 알고리즘 대회
📌2021/01/23 작성 코드
# 입력
n, k = map(int, input().split())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
# a, b 정렬
a.sort() # 오름차순
b.sort(reverse=True) # 내림차순
# 같은 idx의 값 비교
# 종료 조건: idx >= n or k == 0
idx = 0
while idx < n and k > 0:
if a[idx] < b[idx]:
# swap
a[idx], b[idx] = b[idx], a[idx]
k -= 1 # 바꿔치기 횟수 1 감소
idx += 1 # idx 이동
# 출력
print(sum(a))
💭 아이디어
🤓 문제 해설
문제를 해결하기 위한 기본 아이디어는 매번 배열 A에서 가장 작은 원소를 골라서, 배열 B에서 가장 큰 원소와 교체를 하는 것이다.
단, 배열 A에서 가장 작은 원소가 배열 B에서 가장 큰 원소보다 작을 때에만 교체를 수행해야 한다. 이러한 과정을 K번 반복하면 원하는 정답을 얻을 수 있다.
배열 A의 원소를 오름차순으로 정렬하고 배열 B의 원소를 내림차순으로 정렬한다. 그리고 두 배열의 원소를 가장 첫 번째 인덱스부터 차례대로 비교하면서 A의 원소가 B의 원소보다 작을 때에만 교체를 수행한다.
이때 문제에서는 두 배열의 원소가 최대 100,000개까지 입력될 수 있으므로 O(NlogN)을 보장하는 정렬 알고리즘을 이용해야 한다.
🤓 답안 예시
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의 원소보다 한 번 크거나 같아지면 그 뒤로 계속 크거나 같아질 것이기 때문이다! 이런 센스! 부럽군
Author And Source
이 문제에 관하여([PART2] 6-3(정렬): 두 배열의 원소 교체), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@eastgloss0330/PART2-6-3정렬-두-배열의-원소-교체저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)