> 문제설명
xx 회사의 2xN명의 사원들은 N명씩 두 팀으로 나눠 숫자 게임을 하려고 합니다. 두 개의 팀을 각각 A팀과 B팀이라고 하겠습니다. 숫자 게임의 규칙은 다음과 같습니다.
- 먼저 모든 사원이 무작위로 자연수를 하나씩 부여받습니다.
- 각 사원은 딱 한 번씩 경기를 합니다.
- 각 경기당 A팀에서 한 사원이, B팀에서 한 사원이 나와 서로의 수를 공개합니다. 그때 숫자가 큰 쪽이 승리하게 되고, 승리한 사원이 속한 팀은 승점을 1점 얻게 됩니다.
- 만약 숫자가 같다면 누구도 승점을 얻지 않습니다.
전체 사원들은 우선 무작위로 자연수를 하나씩 부여받았습니다. 그다음 A팀은 빠르게 출전순서를 정했고 자신들의 출전 순서를 B팀에게 공개해버렸습니다. B팀은 그것을 보고 자신들의 최종 승점을 가장 높이는 방법으로 팀원들의 출전 순서를 정했습니다. 이때의 B팀이 얻는 승점을 구해주세요.
A 팀원들이 부여받은 수가 출전 순서대로 나열되어있는 배열 A와 i번째 원소가 B팀의 i번 팀원이 부여받은 수를 의미하는 배열 B가 주어질 때, B 팀원들이 얻을 수 있는 최대 승점을 return 하도록 solution 함수를 완성해주세요.
제한사항
A와B의 길이는 같습니다.A와B의 길이는1이상100,000이하입니다.A와B의 각 원소는1이상1,000,000,000이하의 자연수입니다.
> 문제풀이
BruteForce로 풀어야 하나 생각하다가 당연히 O(n^2)일 것 같아
다른방법을 생각해보았다..
아이디어는 모든 점수를 하나의 리스트에 넣어서 오름차순으로 정렬하고, 스택을 활용하여 리스트 객체가 B일때는 스택에 추가, A일때는 스택과 비교하여 answer를 +1해 주었다.
여기서 문제는 효율성과 2(B) 2(B) 2(A) 2(A) 일 경우 2를 리턴하는 불상사가 일어날 수 있다.
따라서 효율성 문제는 정렬에서 PriorityQueue을 사용하였고,
점수 내림차순, 그리고 A가 B보다 먼저 오게 정렬하면 문제를 해결할 수 있다.
import java.util.*;
class Solution {
static class Pos{
char c;
int v;
public Pos(char c, int v){
this.c = c;
this.v = v;
}
}
public int solution(int[] A, int[] B) {
Queue<Pos> pq = new PriorityQueue<>(new Comparator<Pos>() {
@Override
public int compare(Pos o1, Pos o2) {
if(o2.v == o1.v) {
return o1.c - o2.c;
}
else return o2.v - o1.v;
}
});
for(int i=0; i<A.length; i++){
pq.add(new Pos('A',A[i]));
}
for(int i=0; i<B.length; i++){
pq.add(new Pos('B',B[i]));
}
Stack<Pos> st = new Stack<>();
int answer = 0;
while(!pq.isEmpty()) {
Pos p = pq.poll();
if(p.c == 'B') st.add(p);
else {
if(!st.isEmpty()) {
st.pop();
answer++;
}
}
}
return answer;
}
}
문제 출처 : Programmers
PREVIOUS[프로그래머스] Level3 - 셔틀버스