[코딩테스트] 백준 1946번 신입사원 - JAVA

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

Problem 💻

문제

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

 

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

 

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

입력

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

출력

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

 

 

Solution 💡

입력 값으로 점수가 아닌 서류와 면접의 등수가 주어진다.

 

모든 지원자들  (서류 등수, 면접 등수) 형태로 저장해야한다. 이차원 배열을 이용해 (a[i][0], a[i][1]) 형태로 저장했다.

 

지원자가 선발되기 위해서는 서류 심사 성적이나 면접 성적 중 하나는 반드시 이전 선발자보다 좋아야한다.

 

지원자의 합격기준을 만들기 위해 우선 서류 성적을 오름차순 정렬을 하였다. 그럼 서류 등수 1등은 무조건적으로 합격이 된다. 그리고 서류는 이미 정렬이 되었기때문에 다음 지원자들의 서류 성적을 고려하지 않고 면접 성적만 기준을 삼으면 된다.

 

면접 성적의 첫 기준은 서류 1등의 값이 되며, 다음 루프를 돌며 자신보다 높은 성적(좋은 성적을 의미)를 만나면

applicants[i][1] < minInterviewScore 합격자이며, 면접 기준 값을 갱신해준다.

 

면접의 값을 갱신해여 현재까지 선발된 지원자들의 "최소 면접 성적" 기준을 유지하는 것이 중요하다.

갱신된 면접값은 다음 지원자와의 비교에서 정확한 판단이 가능하게 해준다.

 

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());

        StringBuilder sb = new StringBuilder();
        while (t --> 0) {
            int n = Integer.parseInt(br.readLine());
            int[][] applicants = new int[n][2];

            for (int i = 0; i < n; i++) {
                StringTokenizer st = new StringTokenizer(br.readLine());
                applicants[i][0] = Integer.parseInt(st.nextToken());
                applicants[i][1] = Integer.parseInt(st.nextToken());
            }

            // 성적 기준 오름차순 정렬
            Arrays.sort(applicants, (a, b) -> a[0] - b[0]);
            int minInterviewScore = applicants[0][1]; // 기준이 될 면접 순위
            int count = 1; // 첫 번째 사람은 반드시 선발
            // 면접 성적 기준으로 선발
            for (int i = 1; i < n; i++) {
                if (applicants[i][1] < minInterviewScore) {
                    count++;
                    minInterviewScore = applicants[i][1];
                }
            }

            sb.append(count + "\n");
        }

        System.out.println(sb);
    }
}

 

 

멘토링 시간에 나온 답변 중 하나인데 위와 같은 정렬 문제인 TreeSet을 사용하는 것이 좋은 방법 중 하나가 될 수 있다고 하셨다.

TreeSet은 이진 검색 트리를 기반으로 하며, 자동으로 정렬과 중복 방지를 제공한다. 위 같은 문제에서 활용하면 데이터를 삽입하면서 정렬도 가능하니 추가정렬이 필요없게 된다.

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

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()); // 테스트 케이스 수
        
        StringBuilder sb = new StringBuilder();

        while (t-- > 0) {
            int n = Integer.parseInt(br.readLine()); // 지원자 수
            TreeSet<int[]> applicants = new TreeSet<>((a, b) -> a[0] - b[0]); // 서류 성적 기준 정렬
            
            // 지원자 정보 입력
            for (int i = 0; i < n; i++) {
                StringTokenizer st = new StringTokenizer(br.readLine());
                int docScore = Integer.parseInt(st.nextToken());
                int interviewScore = Integer.parseInt(st.nextToken());
                applicants.add(new int[]{docScore, interviewScore});
            }
            
            int count = 0;
            int minInterviewScore = Integer.MAX_VALUE; // 현재까지의 최소 면접 성적
            
            // TreeSet을 순회하며 선발
            for (int[] applicant : applicants) {
                if (applicant[1] < minInterviewScore) { // 면접 성적이 더 좋을 경우 선발
                    count++;
                    minInterviewScore = applicant[1]; // 최소 면접 성적 갱신
                }
            }
            
            sb.append(count).append("\n");
        }
        
        System.out.print(sb);
    }
}