두 배열의 원소 교체

문제

N, K, 그리고 배열 A와 B의 정보가 주어졌을 때, 최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력하는 프로그램을 작성하시오.

 

 

 

 

 

입력

• 첫 번째 줄에 N, K가 공백으로 구분되어 입력된다. (1 <=N <= 1000000 <=k <=N)
• 두 번째 줄에 배열 A의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10000000보다 작은 자연수이다.
• 세 번째 줄에 배열 B의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10000000보다 작은 자연수입니다.

 

출력

최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력한다.

 

입력 예시

5 3
1 2 5 4 3
5 5 6 6 5

출력 예시

26

풀이 방법

배열 두 개에 값을 넣고

정렬을 한다.

한 개는 오름차순, 한개는 내림차순으로 정렬을 하여 앞의 값을 비교해 arrray1의 값이 array2의 값보다 작으면 

둘의 원소를 바꾸고 아니면 break 해 반복문을 탈출시킨다.

내 코드

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

array1=list(map(int,input().split()))
array2=list(map(int,input().split()))

array1.sort()
array2.sort(reverse=True)

for i in range(k):
    if array1[i]<array2[i]:
        array1[i],array2[i]=array2[i],array1[i]
    
    else:
        break

print(sum(array1))

Solution

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

느낀 점

. sort(reverse=True)

역순으로 정렬