정렬 두 배열의 원소 교체

jaegeunsong97·2023년 3월 2일
0
post-thumbnail



import java.util.*;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        // N과 K를 입력받기
        int n = sc.nextInt();
        int k = sc.nextInt();

        // 배열 A의 모든 원소를 입력받기
        Integer[] a = new Integer[n];
        for (int i = 0; i < n; i++) a[i] = sc.nextInt();
        
        // 배열 B의 모든 원소를 입력받기
        Integer[] b = new Integer[n];
        for (int i = 0; i < n; i++) b[i] = sc.nextInt();
        
        // 배열 A는 오름차순 정렬 수행
        Arrays.sort(a);
        // 배열 B는 내림차순 정렬 수행
        Arrays.sort(b, Collections.reverseOrder());
        
         // 첫 번째 인덱스부터 확인하며, 두 배열의 원소를 최대 K번 비교 
         for (int i = 0; i < k; i++) {
            // A의 원소가 B의 원소보다 작은 경우
            if (a[i] < b[i]) {
                // 두 원소를 교체
                int temp = a[i];
                a[i] = b[i];
                b[i] = temp;
            }
            // A의 원소가 B의 원소보다 크거나 같을 때, 반복문을 탈출
            else break;
        }
        
        // 배열 A의 모든 원소의 합을 출력
        long result = 0;
        for (int i = 0; i < n; i++) {
            result += a[i];
        }
        System.out.println(result);
    }

}
import java.io.*;
import java.util.*;

public class Main{
  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()); // 5
    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());

    for (int i = 0; i < K; i++) {
      if (A[i] < B[i]) {
        Integer temp = A[i]; // int temp = A[i] 가능
        A[i] = B[i];
        B[i] = temp;
      } else {
        continue;
      }
    }

    int answer = 0;
    for (int num: A) {
      answer += num;
    }
    System.out.println(answer);
  }
}
profile
현재 블로그 : https://jasonsong97.tistory.com/

0개의 댓글