숫자게임(프로그래머스-level3)

안녕하세요 강민성입니다.

이번에 풀 문제는 2018서머코딩 알고리즘 문제 중 하나 입니다.

대회에서 나오는 문제들을 보면 매번 저런 문제를 어떻게 만들어내는지 그저 신기할 따름입니다.

문제를 이해하고 코드를 짜는 것도 적지 않은 시간이 걸리곤 하는데...

요즘 주변에 취업원서 넣고 코딩테스트 보러 가는 사람들이 많아서 그런지 알고리즘의 중요성을 새삼 다시 깨닫습니다.

1차에서 떨어지는 경우도 비일비재 하더라고요.

그러면 문제 보도록 하겠습니다.

[ 문제 ]

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점입니다.

문제에 여러가지 제한 조건이 많아 처음에는 '어?' 했는데 예시를 보니까 쉽게 이해가 가더군요.

바로 코드를 짤 수 있었습니다.

def solution(A, B):
 
    point = 0
 
    if min(A) >= max(B): # B의 최대값이 A의 최솟값보다도 작거나 같으면 
 
        return 0 # 한 번도 이길 수 없다.
 
    for i in range(len(A)): # A의 길이 만큼 반복을 해서
 
            if max(A) < max(B): # B의 최대값이 A의 최대값보다 더 크면
 
            point = point + 1 # point 값을 1 올려주고
 
            A.remove(A[A.index(max(A))]) # A에서의 최대값을 제거해주고
 
            B.remove(B[B.index(max(B))]) # B에서의 최대값을 제거해준다.
 
        else: # A의 최대값이 더 크면
 
            A.remove(A[A.index(max(A))]) # A 의 최대값만을 제거해준다.
 
    answer = point # point 값이 결과가 된다.
 
    return answer

방식은 다음과 같습니다.

A 에서의 최대값과 B에서의 최대값을비교해준 후,

B가 더 크다면 둘 모두 지워주고 point를 하나 올려줍니다.

A에서의 최대값이 더 클 경우에는 A에서만 지워줘야 하는데요, 

그 이유는 위의 예시를 보며 이해해보겠습니다.

A = [ 5 , 1 , 3 , 7 ]

B = [ 2 , 2 , 6 , 8] 일 경우, 

최대값을 두 번 제거하면 

A = [ 1 , 3 ]

B = [ 2 , 2 ]  다음과 같이 남게 되고 point는 2만큼 쌓였습니다.

이 때 B 가 A 를 한 번 더 이길 수 있지만

최대값만을 비교하면 A의 최대값이 더 크므로 문제가 생기죠.

그래서 A의 최대값을 제거해줘야 하는 겁니다.

이렇게 짠 결과, 정답은 모두 맞았지만 시간복잡도에서 오류가 뜨더라고요.

그래서 수정한 풀이는 다음과 같습니다.

def solution(A, B):
 
    answer = 0
 
    point = 0 
 
    sortA = sorted(A) # A를 오름차순으로 정렬
 
    sortB = sorted(B) # B를 오름차순으로 정렬
 
    for i in sortA: # A 안에 수들을 순회하며
 
        for j in sortB: # B 안에 수들을 순회하며
 
            if i<j: # B의 값이 더 크면
 
                point+=1 # point 를 1 증가시키고
 
                sortB.remove(j) # 그 값을 B에서 지워준 후
 
                break # 반복문을 끝낸다.
 
    answer = point
 
    return answer

위의 풀이와 같은 논리로 해결을 합니다.

하지만 차이가 있다면 첫 번째에서는 max 와 min 을 이용해서 최대, 최소 값을 찾아주는데

이번 풀이에서는 sorting 을 함으로 인해서 미리 크기 순으로 정렬을 시켜놓습니다.

이 방법을 통해 시간효율을 크게 늘릴 수 있었습니다.

마무리

파이썬으로 코드를 짜다보니 시간복잡도를 생각해주기가 더 어렵다는 생각이 들었습니다.

크게 복잡하지 않은 코드같은데 간단한 함수 하나하나가 시간을 잡아먹어서 오류가 떴습니다.

항상 시간복잡도를 생각해주는게 어려워 많이 노력해야겠습니다.