[백준 7795] 먹을 것인가 먹힐 것인가 (Java) - 투포인터

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

먹을 것인가 먹힐 것인가

문제

심해에는 두 종류의 생명체 A와 B가 존재한다. A는 B를 먹는다. A는 자기보다 크기가 작은 먹이만 먹을 수 있다. 예를 들어, A의 크기가 {8, 1, 7, 3, 1}이고, B의 크기가 {3, 6, 1}인 경우에 A가 B를 먹을 수 있는 쌍의 개수는 7가지가 있다. 8-3, 8-6, 8-1, 7-3, 7-6, 7-1, 3-1.

 

두 생명체 A와 B의 크기가 주어졌을 때, A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 A의 수 N과 B의 수 M이 주어진다. 둘째 줄에는 A의 크기가 모두 주어지며, 셋째 줄에는 B의 크기가 모두 주어진다. 크기는 양의 정수이다. (1 ≤ N, M ≤ 20,000) 

출력

각 테스트 케이스마다, A가 B보다 큰 쌍의 개수를 출력한다.

예제 입력 1 복사

2
5 3
8 1 7 3 1
3 6 1
3 4
2 13 7
103 11 290 215

예제 출력 1 복사

7
1

---

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args)throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(br.readLine()); // 테스트 케이스 개수

        // 테스트 시작
        for(int t=0; t<T; t++){
            StringTokenizer st = new StringTokenizer(br.readLine());
            int N = Integer.parseInt(st.nextToken()); // A개수
            int M = Integer.parseInt(st.nextToken()); // B개수

            int[] A = new int[N];
            int[] B = new int[M];

            // A 배열 입력받기
            st = new StringTokenizer(br.readLine());
            for(int i=0; i<N; i++){
                A[i] = Integer.parseInt(st.nextToken());
            }
            Arrays.sort(A); // 오름차순 정렬

            // B 배열 입력받기
            st = new StringTokenizer(br.readLine());
            for(int i=0; i<M; i++){
                B[i] = Integer.parseInt(st.nextToken());
            }
            Arrays.sort(B); // 오름차순 정렬

            int j = 0;
            int count = 0;
            for(int i=0; i<N; i++){
                while (j<M && B[j] < A[i]){
                    j++;
                }
                count += j;
            }

            System.out.println(count);
        }
    }
}

• 투포인터를 활용한 문제
• B < A 인 모든 조합을 찾는 것
• A, B배열의 정보를 받아서 오름차 순 정렬해준다.
• i=0 ~ N까지 순회하면서 B[j] < A[i] 를 만족할때까지 j포인터를 증가시켜준다.
• 만족하지 않는 시점이 오면 중단하고 count값에 더한다.