본문 바로가기
알고리즘

두 배열의 원소 교체

by 토니짱 2022. 9. 16.

*토니는 두 개의 배열 A,B를 가지고 있다. 두 배열은 N개의 원소로 구성되어 있으며, 배열의 원소는 모두 자연수이다.

*토니는 최대 K번의 바꿔치기 연산을 수행할 수 있는데, 바꿔치기 연산이란 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 것을 말한다.

*최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록 하는 것이다.

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

 

입력 조건

-첫 번째 줄에 N,K가 공백을 기준으로 구분되어 입력됩니다.(1 <= N <= 100,000, 0 <= K <= N)

-두 번째 줄에 배열 A의 원소들이 공백을 기준으로 구분되어 입력됩니다. 모든 원소는 10,000,000보다 작은 자연수입니다.

-세 번째 줄에 배열 B의 원소들이 공백을 기준으로 구분되어 입력됩니다. 모든 원소는 10,000,000보다 작은 자연수입니다.

 

출력 조건

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

 

입력예시

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

 

출력 예시

26

 

나의 생각

A배열의 합만 최댓값으로 만들면 되니까 A는 오름차순 정렬하고 B는 내림차순 정렬하여 K개만큼 비교한 뒤에 원소를 바꾸면 최댓값이 나올 것 같다.

 

N, K = map(int, input().split())
arrA = list(map(int,input().split(' ')))
arrB = list(map(int,input().split(' ')))

arrA.sort()
arrB.sort(reverse = True)

for i in range(K):
  if(arrA[i] < arrB[i]):
    arrA[i],arrB[i] = arrB[i],arrA[i]
  else:break

print(sum(arrA))
반응형

'알고리즘' 카테고리의 다른 글

단지번호붙이기 - 2667 (DFS & BFS)  (0) 2022.09.16
[level 1] 신고 결과 받기 - 92334  (0) 2022.09.16
미로 탈출 BFS문제  (4) 2022.09.15
음료수 얼려먹기 DFS문제  (0) 2022.09.15
왕실 나이트 문제  (0) 2022.09.15