boj7795 먹을 것인가 먹힐 것인가_java

링크

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보다 큰 쌍의 개수를 출력한다.

나의 풀이

• 먹는 주체를 s 라고 하고 먹을 타겟을 t 라고 하자.
  s와 t를 0을 초기값으로 시작해서 s==t 이거나 s<t 가 될때, s++ 을 해주고, s>t 일때는 t++ 를 해주었다. 이때, t==M 이 되는 순간부터는 answer 에다가 t값을 저장하고 s++ 을 해준다.
• 핵심 코드

  Arrays.sort(A);
              Arrays.sort(B);
              int answer =0;
              int s = 0;
              int t = 0;
              while(s<N){
                  if(t==M){
                      answer += t;
                      s+=1;
                  }else{
                      if(A[s]<= B[t]){
                          s+=1;
                          answer += t;
                      }
                      else{
                          t+=1;
                      }
                  }
              }

• 전체 코드

  package Baekjoon.java.silver;
  
  import java.io.InputStreamReader;
  import java.io.*;
  import java.util.*;
  
  public class boj7795 {
      public static void main(String[] args) throws Exception{
          BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
          StringBuilder sb = new StringBuilder();
          int T = Integer.parseInt(br.readLine());
          for (int tc = 0; tc < T; tc++) {
              StringTokenizer st = new StringTokenizer(br.readLine());
              int N = Integer.parseInt(st.nextToken());
              int M = Integer.parseInt(st.nextToken());
  
              int[] A = new int[N];
              int[] B = new int[M];
  
              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 < M; i++) {
                  B[i] = Integer.parseInt(st.nextToken());
              }
              Arrays.sort(A);
              Arrays.sort(B);
              int answer =0;
              int s = 0;
              int t = 0;
              while(s<N){
                  if(t==M){
                      answer += t;
                      s+=1;
                  }else{
                      if(A[s]<= B[t]){
                          s+=1;
                          answer += t;
                      }
                      else{
                          t+=1;
                      }
                  }
              }
  
              sb.append(answer+"\n");
          }
  
          System.out.println(sb.toString());
      }
  }

결과