알고리즘 공부/정렬(Sort) | 분류

[프로그래머스] 숫자게임

kdhoooon 2021. 6. 10. 16:40

문제


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 이하의 자연수입니다.

 

 

 

 

풀이


처음에 완전탐색으로 모든 경우의 수를 찾아서 그중에 최고 점수를 찾았다.

당연하게도 시간초과가 났다.

 

그래서 가장 좋은 점수를 얻을 수 있는 방법을 생각했다.

2가지 조건이 있으면 가장 좋은 점수를 얻을 수 있다.

  1. 이길 수 있다면 가장 가까운 큰수를 낸다.
  2. 이길 수 없다면 가장 작은 수를 낸다.

이 두가지 경우의 수를 생각하고 문제를 먼저 풀어보면 좋을 것 같다. 어려운 알고리즘적인 스킬이 들어가는 것이 아니라 경우의 수를 고려하여 문제를 풀면 되기 때문에 고민을 해보고 풀면 좋을 것 같다.

 

알고리즘 순서

더보기
  1.  A, B를 오름차순으로 정렬한다.
  2.  A, B의 가장 뒤 부터 비교하여 B가 A보다 크면 하나 다음 수를 비교해준다.
  3.  만약 A 가 더 크다면 B는 가만히 있고 A 만 다음수로 넘어간다. 이경우 가장 작은 수를 냈다고 가정한다.
  4.  A의 숫자가 0번째 숫자면 반복문을 끝내고 그때의 점수를 return 한다.

 

 

<전체코드>

더보기
import java.util.*;

class Solution {

    public int solution(int[] A, int[] B) {
        int answer = 0;
        
        Arrays.sort(A);
        Arrays.sort(B);
        
        int j = B.length - 1;
        for(int i = A.length - 1 ; i >= 0 ; i--){
            if(A[i] < B[j]){
                answer++;
                j--;
            }
        }
        
        return answer;
    }   
}