동빈이는 두 개의 배열 A와 B를 가지고 있다. 두 배열은 N개의 원소로 구성되어 있으며, 배열의 원소는 모두 자연수이다
동빈이는 최대 K 번의 바꿔치기 연산을 수행할 수 있는데, 바꿔치기 연산이란 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 것을 말한다
동빈이의 최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록 하는 것이며, 여러분은 동빈이를 도와야한다
N, K, 그리고 배열 A와 B의 정보가 주어졌을 때, 최대 K 번의 바꿔치기 연산을 수행하여 만들 수 있는
배열 A의 모든 원소의 합의 최댓값을 출력하는 프로그램을 작성하라
예를 들어 N = 5, K = 3이고, 배열 A와 B가 다음과 같다고 해보자
배열 A = [1, 2, 5, 4, 3]
배열 B = [5, 5, 6, 6, 5]
이 경우, 다음과 같이 세 번의 연산을 수행할 수 있다
연산 1) 배열 A의 원소 '1'과 배열 B의 원소 '6'을 바꾸기
연산 2) 배열 A의 원소 '2'와 배열 B의 원소 '6'을 바꾸기
연산 3) 배열 A의 원소 '3'과 배열 B의 원소 '5'를 바꾸기
세 번의 연산 이후 배열 A와 배열 B의 상태는 다음과 같이 구성될 것이다
배열 A = [6, 6, 5, 4, 5]
배열 B = [3, 5, 1, 2, 5]
이때 배열 A의 모든 원소의 합은 26이 되며, 이보다 더 합을 크게 만들 수는 없다
입력
출력
최대 K번 바꿔치기 연산을 수행해서 가장 최대의 합을 갖는 A의 모든 원소 값의 합을 출력
입력 예시
5 3
1 2 5 4 3
5 5 6 6 5
출력 예시
26
A배열은 오름차순
B배열은 내림차순.
A배열의 원소가 B의 배열의 원소보다 작을때만 스위치 해준다.
이걸 K번 수행하면 된다.
- Arrays.sort 내림차순 방법
Arrays.sort(B, Collections.reverseOrder());
- Arrays.sort를 내림차순 하기 위해서는 배열 선언이 Integer로 되어야 한다.
Integer A[] = new Integer[N]; Integer B[] = new Integer[N];
- A의 원소가 B의 원소보다 크다면 바로 반복문을 탈출한다.
오름차순, 내림차순으로 이미 정렬을 했기 때문에 A의 원소가 클 경우 계속 크기 때문.for(int i = 0; i < K; i++) { if(A[i] < B[i]) { A[i] = B[i]; } else { break; } }
import java.util.*;
import java.io.*;
public class 정렬_두배열의원소교체 {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine(), " ");
int N = Integer.parseInt(st.nextToken());
int K = Integer.parseInt(st.nextToken());
Integer A[] = new Integer[N];
Integer B[] = new Integer[N];
st = new StringTokenizer(br.readLine(), " ");
for(int i = 0; i < N; i++) {
A[i] = Integer.parseInt(st.nextToken());
}
st = new StringTokenizer(br.readLine(), " ");
for(int i = 0; i < N; i++) {
B[i] = Integer.parseInt(st.nextToken());
}
Arrays.sort(A);
Arrays.sort(B, Collections.reverseOrder());
//int index = 0;
for(int i = 0; i < K; i++) {
if(A[i] < B[i]) {
A[i] = B[i];
//index++;
} else {
break;
}
}
int sum = 0;
for(int i = 0; i < N; i++) {
sum += A[i];
}
System.out.println(sum);
}
}