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

[프로그래머스][Python] 숫자 게임, level 3

문제

문제 설명

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

입출력 예

A	B	result
[5,1,3,7]	[2,2,6,8]	3
[2,2,2,2]	[1,1,1,1]	0

입출력 예 설명

• 입출력 예 #1
  
  A 팀은 숫자 5를 부여받은 팀원이 첫번째로 출전하고, 이어서 1,3,7을 부여받은 팀원들이 차례대로 출전합니다.
  B 팀원들을 4번, 2번, 3번, 1번의 순서대로 출전시킬 경우 팀원들이 부여받은 숫자들은 차례대로 8,2,6,2가 됩니다. 그러면, 첫 번째, 두 번째, 세 번째 경기에서 승리하여 3점을 얻게 되고, 이때가 최대의 승점입니다.

• 입출력 예 #2
  B 팀원들을 어떤 순서로 출전시켜도 B팀의 승점은 0점입니다.

---

접근

문제의 접근 방향은 맞았으나,
구현에 있어서 너무 어렵게 생각했다.

B의 승점을 가장 크게 하기 위해서는
A의 각 원소보다 큰 수 중에서 최소값 을 같은 인덱스에 할당해주어야한다고 생각했다.

1

나는 어떤 수와 가장 가까운 큰 수를 찾기 위해서는
uppder bound를 사용해야 한다고 생각하고,
리스트 B를 정렬한 뒤
리스트 A의 각 원소를 순회하면서
각 원소 A[i]와 가까운 큰 수를 찾기위해 매 순회마다 B에서 uppder bound를 통해 조건에 맞는 인덱스를 찾아주었다.

2

최초에는 사용한 인덱스의 인덱스 제거를 위해서 리스트를 슬라이싱 했는데,
중간에 있는 값을 삭제하기 위해서 B = B[:index] + [index+1:] 와 같은 식으로
매 순회마다 리스트를 슬라이싱하고 생성하다 보니 효율성 검사를 통과하지 못했다.

3

그래서 객체를 다시 생성하는 것 보다는
사용한 인덱스를 다른 배열에 표시하고,
uppder bound의 결과로 나온 인덱스가 이미 사용된 경우에는
그것보다 큰 인덱스를 찾기위해 배열을 순서대로 순회하는 로직을 추가했다.

이 코드는 통과했고, 그 코드는 아래와 같다.

def solution(A, B):
    answer = 0
    min_idx = 0
    B.sort()
    used = [False] * (len(A))

    for i in range(len(A)):
        idx = upper_bound(B, A[i])
        idx = find_min(used, idx, len(A))
        if idx <= len(B)-1 and A[i] < B[idx]:
            used[idx] = True
            answer += 1
        else:
            min_idx = find_min(used, min_idx, len(A))
            used[min_idx] = True
        # print(f'A:B = {A[i]}:{B[idx]}')
    return answer

def upper_bound(array:list, target:int):
    s = 0
    e = len(array)-1
    m = 0
    while s < e:
        m = (s+e)//2
        if array[m] > target:
            e = m
        else:
            s = m + 1
    if array[e] == target:
        return e + 1
    else:
        return e
    
def find_min(used:list, cur:int, array_size:int):
    for i in range(cur, array_size):
        if not used[i]:
            return i
    return array_size

하지만 뭔가 문제의 단순함에 비해 로직이 너무 복잡하다고 느껴
다른 분들의 풀이를 확인해보았다.

4

.. 배열을 단순히 순회 하면서도
A의 각 원소보다 큰 수 중에서 최소값 을 찾을 수 있었다.
그냥 A, B 두 배열을 모두 오름차순 정렬하고
배열 A를 기준으로 순회하며
A[index_a] < B[index_b] 일 때만 index_b의 값을 1 증가시켜주면 되었다..

두 배열 모두 오름차순으로 정렬되어있기 때문에
해당 원소보다 큰 수 중 최소값 을 항상 만족할 수 있다.

5

자꾸 문제에 갇히는 느낌이다.
해당 원소보다 큰 수 중 최소값 을 찾는 방법을 고민하기 보다는
배열 A에 배열 B의 원소들을 매칭시켜야 한다는 생각 때문에
특정한 인덱스를 찾아주어야 한다고 생각했고,
이로 인해 사용된 인덱스의 사용 여부를 처리해야 한다는 생각에 매몰되었다.

해당 원소보다 큰 수 중 최소값 을 찾는것이 문제의 핵심이라고 판단되었다면
문제의 내용 자체보다는 거기에 집중하는 습관이 필요하다고 느낀다.

코드

최종적으로 두 배열을 정렬하고 $O(n)$으로 풀이한 코드이다.
굉장히 간결하다.

def solution(A, B):
    answer = 0
    A.sort()
    B.sort()
    i, j = 0, 0
    while i < len(A) and j < len(B):
        if A[i] < B[j]:
            answer += 1
            i += 1
        j += 1
    return answer