[코딩테스트] 신입 사원

문제

언제나 최고만을 지향하는 굴지의 대기업 진영 주식회사가 신규 사원 채용을 실시한다. 인재 선발 시험은 1차 서류심사와 2차 면접시험으로 이루어진다. 최고만을 지향한다는 기업의 이념에 따라 그들은 최고의 인재들만을 사원으로 선발하고 싶어 한다.

그래서 진영 주식회사는, 다른 모든 지원자와 비교했을 때 서류심사 성적과 면접시험 성적 중 적어도 하나가 다른 지원자보다 떨어지지 않는 자만 선발한다는 원칙을 세웠다. 즉, 어떤 지원자 A의 성적이 다른 어떤 지원자 B의 성적에 비해 서류 심사 결과와 면접 성적이 모두 떨어진다면 A는 결코 선발되지 않는다.

이러한 조건을 만족시키면서, 진영 주식회사가 이번 신규 사원 채용에서 선발할 수 있는 신입사원의 최대 인원수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 테스트 케이스의 개수 T(1 ≤ T ≤ 20)가 주어진다. 각 테스트 케이스의 첫째 줄에 지원자의 숫자 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개 줄에는 각각의 지원자의 서류심사 성적, 면접 성적의 순위가 공백을 사이에 두고 한 줄에 주어진다. 두 성적 순위는 모두 1위부터 N위까지 동석차 없이 결정된다고 가정한다.

출력

각 테스트 케이스에 대해서 진영 주식회사가 선발할 수 있는 신입사원의 최대 인원수를 한 줄에 하나씩 출력한다.

예제 입력 1

2
5
3 2
1 4
4 1
2 3
5 5
7
3 6
7 3
4 2
1 4
5 7
2 5
6 1

예제 출력 1

4
3

생각

• 일단 첫 번째 기준인 서류를 기준으로 지원자들을 정렬한 뒤, 첫 번째 (즉 서류 1등)의 면접 성적을 기준으로 이후 지원자들이 면접에서 더 뛰어난지를 판단한다
  • 만약 면접에서 더 뛰어난 사람이 등장했을 때는 minInterviewRank를 업데이트한다

코드

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

public class NewRecruit {

    public int getMaxRecruit(int N, int[][] arr) {
        Arrays.sort(arr, (a, b) -> Integer.compare(a[0], b[0]));

        int count = 1;
        int minInterviewRank = arr[0][1];

        for (int i = 1; i < N; i++) {
            if (arr[i][1] < minInterviewRank) {
                count++;
                minInterviewRank = arr[i][1];
            }
        }

        return count;
    }

    public static void main(String[] args) throws IOException {
        NewRecruit T = new NewRecruit();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int K = Integer.parseInt(br.readLine());
        StringBuilder sb = new StringBuilder();

        for (int j = 0; j < K; j++) {
            int N = Integer.parseInt(br.readLine());
            int[][] arr = new int[N][2];
            for (int i = 0; i < N; i++) {
                StringTokenizer st = new StringTokenizer(br.readLine());
                arr[i][0] = Integer.parseInt(st.nextToken());
                arr[i][1] = Integer.parseInt(st.nextToken());
            }
            sb.append(T.getMaxRecruit(N, arr)).append("\n");
        }

        System.out.print(sb.toString());
        br.close();
    }
}

정리

Arrays.sort 메서드

• java 배열을 정렬하기 위해 사용되며, Comparator 인터페이스를 구현하는 객체를 전달하여 정렬 기준을 명시할 수 있다

• 예시 : 성, 그리고 성이 같은 경우 이름을 기준으로 오름차순 정렬하고 싶을 때

String[][] people = {
    {"Smith", "John"},
    {"Doe", "Jane"},
    {"Smith", "Alex"},
    {"Johnson", "Emily"}
};

Arrays.sort(people, (a, b) -> {
    int lastNameComp = a[0].compareTo(b[0]);
    if (lastNameComp != 0) {
        return lastNameComp; // 성이 다르면 성으로 정렬
    } else {
        return a[1].compareTo(b[1]); // 성이 같으면 이름으로 정렬
    }
});

• 정렬 후 모습

{
    {"Doe", "Jane"},
    {"Johnson", "Emily"},
    {"Smith", "Alex"},
    {"Smith", "John"}
}

참조

• 이차원 이상의 배열이 동일한지 확인할 때는 assertArrayEquals 로 확인하는 대신 아래와 같이 Arrays.deepEquals() 메서드를 사용해서 확인해야 한다

assertTrue(Arrays.deepEquals(expected, result));

---

• REF