[프로그래머스] 인사고과 파이썬

https://school.programmers.co.kr/learn/courses/30/lessons/152995

프로그래머스

문제 설명:

완호네 회사는 연말마다 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 합니다.

 

풀이:

저 인센티브를 받지 못하는 사람을 Time complexity 를 O(N^2) 으로 제거하면 시간 초과가 된다. 처음에 이것때문에 애좀 먹었다.

그래서 내가 푼 방법은 우선 score을 근무태도를 내림차순으로 정렬한다. 같은 근무태도 점수에서는 동료 평가 점수를 오름차순으로 정렬한다.

그렇게 하고나면 인센티브를 받지 못하는 사람들을 O(N)으로 거를 수 있다. 전에 근무태도 점수가 높았던 그룹의 동료평가 점수의 최대값보다 작은 애들은 전부 받지 못하게 되니까.

 

코드:

def solution(scores):
    answer = 1

    target = scores[0]
    target_score = sum(scores[0])
    scores.sort(key=lambda x: (-x[0], x[1]))

    threshold = 0
    for score in scores:
        if target[0] < score[0] and target[1] < score[1]:
            return -1
        if threshold <= score[1]:
            if target_score < score[0] + score[1]:
                answer += 1
            threshold = score[1]
    return answer