백준 | 1946번. 신입사원 - 그리디 & 정렬 기준 재정의

728x90

🟦 백준 1946번. 신입사원 - 그리디

문제

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

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

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

입력

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

출력

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


💚나의 풀이 -(1) 그리디 & Sawon 클래스

  • 적어도 하나가 다른 지원자보다 떨어지지 않는 자만 선발한다는 원칙
  • 즉, 둘다 떨어지는 지원자는 탈락시킨다.
  • 그러므로 성적 순 정렬시켜놓고, 선발시킨 '현재'만 기준으로 생각 (그리디)
  • 현재 뽑은 사람의 면접 점수를 기준 점수로 갱신시켜놓고
  • 이 점수보다 뒤쳐지면. 그 지원자는 (이미 성적도 밀렸는데) 면접 까지 뒤쳐지는 지원자이므로 건너뜀

🏓 Sawon 클래스 implements Comparable<Sawon> 구현해서 내부에 compareTo() 메소드 재정의하고 정렬 기준을 return this.성적 - o.성적 오름차순 정렬하도록 바꿨다.

class Sawon implements Comparable<Sawon>{
	int sj;
	int mj;
	Sawon(int sj, int mj){
		this.sj = sj;
		this.mj = mj;
	}
	@Override
	public int compareTo(Sawon o) {
		// TODO Auto-generated method stub
		return this.sj - o.sj;//성적 기준 오름차순 정렬 
	}
}
public class Main {
	//실행 메인 
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner kb = new Scanner (System.in);
		
		int T = kb.nextInt();
		ArrayList<Integer> answer = new ArrayList<>();
		
		for(int t=0; t<T; t++) {
			ArrayList<Sawon> arr = new ArrayList<>();
			int n = kb.nextInt();
			
			for(int i=0; i<n; i++) {
				int a = kb.nextInt();
				int b = kb.nextInt();
				arr.add(new Sawon(a, b));
			}
			Collections.sort(arr);
			
			int cnt = 1;
			int pivot = arr.get(0).mj;//첫 번째 인간 성적 기준으로 두고 
			
			for(int i=1; i<arr.size(); i++) {
				if(pivot < arr.get(i).mj) continue;
				
				cnt++;
				pivot = arr.get(i).mj;
			}
			answer.add(cnt);
			
		}
		
		for(int x : answer) System.out.println(x);
	}
}

🏓 JAVA | 배열 Arrays.sort() 기준 재정의 : Comparator 익명 클래스 구현

🟧 1) 1차원 배열 sort 재정의

//1차원 배열 내림차순 sort - compareTo 사용
Integer[] nums6 = {9,8,1,3,2};
Arrays.sort(nums6, new Comparator<Integer>() {
    @Override
    public int compare(Integer o1, Integer o2) { //두 숫자 비교
        return o1 - o2; //오름차순 
				//return o2 - o1; //내림차순 
    }
});

🟧 2) 2차원 배열 sort 재정의

int[][] arr = new int[][2]{{5,40},{3,50},{1,30},{4,20},{2,10}};

// 1. Comparator 익명 클래스 구현
Arrays.sort(arr, new Comparator<int[]>() {
    //compare() 메소드 재정의 
		@Override
    public int compare(int[] o1, int[] o2) {       
				return o1[0]-o2[0]; // 첫번째 숫자 기준 오름차순 {1,30}{2,10}{3,50}{4,20}{5,40}
        //return o2[0]-o1[0]; // 첫번째 숫자 기준 내림차순 {5,40}{4,20}{3,50}{2,10}{1,30}
        //return o1[1]-o2[1]; // 두번째 숫자 기준 오름차순 {2,10}{4,20}{1,30}{5,40}{3,50}
        //return o2[1]-o1[1]; // 두번째 숫자 기준 내림차순 {3,50}{5,40}{1,30}{4,20}{2,10}
    }
});

💚나의 풀이 - (2) 그리디 & 2차원 A[i][2] 배열

  • 정답은 제대로 나왔는데, 백준에서는 시간 초과가 떴다.
  • 객체로 만들어서 정렬해주는 게 더 나을 듯 하다.
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;
/*신입사원 - 그리디 RE */
public class Main {
	//실행 메인 
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner kb = new Scanner(System.in);
		int T = kb.nextInt();
		ArrayList<Integer> answer = new ArrayList<>();
		for(int t= 0; t<T; t++) {
			int n = kb.nextInt();
			int[][] A = new int[n][2];
			
			for(int i=0; i<n; i++) {
				int a = kb.nextInt();
				int b = kb.nextInt();
				A[i][0] = a;//성적 순위 
				A[i][1] = b; //면접 순위 
			}
			
			Arrays.sort(A, new Comparator<int[]>() {
				@Override
				public int compare(int[] o1, int[] o2) {
					// TODO Auto-generated method stub
					return o1[0]-o2[0];//0번쨰 성적 순위 기준 오름차순 정렬 
				}
			});
			//그리디 - 현재 뽑은 애를 기준으로만 생각
			int cnt = 1;
			int pivot = A[0][1];//면접 첫번쨰 인간 
			
			for(int i=1; i<n; i++) {
				if(pivot < A[i][1]) continue;//면접까지 뒤쳐지는 애는 패스
				cnt++;
				pivot = A[i][1];//면접 접수는 현재 선발자로 갱신
			}
			answer.add(cnt);
		}
		//정답 출력
		for(int x : answer) System.out.println(x);
	}
}
728x90