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

장우솔·2023년 2월 17일
0

알고리즘

목록 보기
10/21
post-thumbnail

핵심 아이디어

매번 배열 A에서 가장 작은 원소를 골라서 배열B에서 가장 큰 원소와 교체한다. 단 A원소가 B의 원소보다 작을 때만 교체를 수행한다.

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의 모든 원소의 합을 출력
profile
공부한 것들을 정리하는 블로그

0개의 댓글