프로그래머스 - 인사고과 (Python)

Lv3, 정렬

문제 설명

완호네 회사는 연말마다 1년 간의 인사고과에 따라 인센티브를 지급합니다. 각 사원마다 근무 태도 점수와 동료 평가 점수가 기록되어 있는데 만약 어떤 사원이 다른 임의의 사원보다 두 점수가 모두 낮은 경우가 한 번이라도 있다면 그 사원은 인센티브를 받지 못합니다. 그렇지 않은 사원들에 대해서는 두 점수의 합이 높은 순으로 석차를 내어 석차에 따라 인센티브가 차등 지급됩니다. 이때, 두 점수의 합이 동일한 사원들은 동석차이며, 동석차의 수만큼 다음 석차는 건너 뜁니다. 예를 들어 점수의 합이 가장 큰 사원이 2명이라면 1등이 2명이고 2등 없이 다음 석차는 3등부터입니다.

각 사원의 근무 태도 점수와 동료 평가 점수 목록 scores이 주어졌을 때, 완호의 석차를 return 하도록 solution 함수를 완성해주세요.

제한 사항

• 1 ≤ scores의 길이 ≤ 100,000
• scores의 각 행은 한 사원의 근무 태도 점수와 동료 평가 점수를 나타내며 [a, b] 형태입니다.
  • scores[0]은 완호의 점수입니다.
  • 0 ≤ a, b ≤ 100,000
• 완호가 인센티브를 받지 못하는 경우 -1을 return 합니다.

입출력 예시

풀이

처음엔 마냥 쉬운 줄 알고 이게 Lv3인가 싶었다.
그런데 정렬 기준을 잡는 부분에서 애를 먹었다.

• a 기준 내림차순 정렬, b기준 오름차순 정렬을 해준다.
  • 예시 기준 [3,2], [3,2], [2,1], [2,2], [1,4] 로 정렬
  • 완호의 점수가 둘 다 작을 경우, -1 리턴
  • 이후, b의 최대값을 기준으로 갱신이 될 때,
    a+b > sum(완호)를 비교해 완호의 순서를 정한다.
    • 이미 a가 작거나 같은데, b가 더 작다면 비교조차 할 필요가 없다.

def solution(scores):
    answer = 1
    me = scores[0]
    scores.sort(key=lambda x : (-x[0], x[1]))
    maxB = 0

    for a, b in scores:
        if me[0] < a and me[1] < b:
            return -1

        if b >= maxB:
            maxB = b
            if a + b > sum(me):
                answer += 1
    return answer

생각보다 어려워서 애를 먹었다...