[백준] 7795번 - 먹을 것인가 먹힐 것인가 [Java]

https://www.acmicpc.net/problem/7795

 

[백준] 7795번 - 먹을 것인가 먹힐 것인가 [Java]

[백준] 7795번 - 먹을 것인가 먹힐 것인가 [Java]

---

1.  아이디어

 

투 포인터 알고리즘을 활용하면 간단하게 해결할 수 있다.

---

2. 문제풀이

 

두 종류의 생명체를 각각 배열에 저장한 후 정렬을 수행한다.

이후 각 배열의 가장 큰 원소 위치에 각각 포인터를 설정하고 해당 포인터를 기준으로 매칭되는 쌍의 개수를 세어줬다.

A 포인터 위치의 생명체가 B 포인터 위치의 생명체보다 크다면 B 포인터의 위치로 더 작은 생명체들의 수를 바로 알 수 있다.

---

3. 코드

 

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));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        int T = Integer.parseInt(br.readLine());

        for (int tc = 1; tc <= T; tc++) {
            st = new StringTokenizer(br.readLine());
            int N = Integer.parseInt(st.nextToken());
            int M = Integer.parseInt(st.nextToken());

            int[] A = new int[N];
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < N; i++) {
                A[i] = Integer.parseInt(st.nextToken());
            }
            Arrays.sort(A);

            int[] B = new int[M];
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < M; i++) {
                B[i] = Integer.parseInt(st.nextToken());
            }
            Arrays.sort(B);

            int pointA = N - 1;
            int pointB = M - 1;
            int sum = 0;
            while (pointA >= 0 && pointB >= 0) {
                if (A[pointA] > B[pointB]) {
                    sum += pointB + 1;
                    pointA--;
                } else {
                    pointB--;
                }
            }

            sb.append(sum).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기