곰퓨타의 SW 이야기

[프로그래머스 level3 숫자게임] 그리디 응용하기 본문

TIL/프로그래머스

[프로그래머스 level3 숫자게임] 그리디 응용하기

곰퓨타 2021. 9. 28. 13:00

해결해야하는 문제는 다음과 같았다.
https://programmers.co.kr/learn/courses/30/lessons/12987

 

코딩테스트 연습 - 숫자 게임

xx 회사의 2xN명의 사원들은 N명씩 두 팀으로 나눠 숫자 게임을 하려고 합니다. 두 개의 팀을 각각 A팀과 B팀이라고 하겠습니다. 숫자 게임의 규칙은 다음과 같습니다. 먼저 모든 사원이 무작위로

programmers.co.kr


아이디어는 다음과 같다.


아이디어를 바탕으로 작성한 코드는 다음과 같다.

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

    return answer
Comments