[정렬] 두 배열의 원소 교체 (실전문제, 이것이 코딩테스트다 with 파이썬)

문제

해설

매번 배열 A에서 가장 작은 원소를 골라서, 배열 B에서 가장 큰 원소와 교체

단, 배열 A에서 가장 작은 원소가 배열 B에서 가장 큰 원소보다 작을 때에만 교체 수행

두 배열의 원소가 최대 100,000개까지 입력 → O(NlogN) 보장하는 정렬 알고리즘 이용

 

풀이

# 변수 : N 두 배열의 원소 개수, K 바꿔치기 연산 횟수
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의 모든 원소의 합을 출력

 

해석

변수 : N 두 배열의 원소 개수, K 바꿔치기 연산 횟수, A B 두 배열

 

1. 두 배열의 원소 개수(N), 바꿔치기 연산 횟수(K) 입력받기

2. 두 배열 A, B 입력받기

3. A의 오름차순, B의 내림차순 정렬

4. A, B의 원소를 비교하고 교체

ㄴ A 원소 > B 원소 : 교체

ㄴ A 원소 <= B 원소 : 반복문 탈출

5. 배열 A의 원소 합 출력