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

link 🔗 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 합니다.

 

---

 

🔥 근무태도와 동료 평가 점수를 기준으로 역순 정렬하여 동료 평가 점수만 비교해서 for문 사용 횟수를 줄이는 것까진 아이디어를 생각해 작성했는데... 뭔가 더 간단하게 작성할 수 있지 않을까 싶었다🤔 scores의 최대 길이가 10만이기에 시간초과가 나지 않을까 했는데, 아니나 다를까 시간초과가...!!

🔥 다른 분의 코드를 보며 정렬을 하되, 근무태도는 역순, 동료 평가는 기본 순서로 정렬하면 이중for문을 활용하지 않아도 된다는 걸 깨달았다!

def solution(scores):
    answer = 1
    wanho = scores[0]
    length = len(scores)
    
    # 근무태도를 기준으로 역순 정렬, 동료 평가를 기준으로 기본 정렬
    scores.sort(key=lambda x: (-x[0], x[1]))
    
    # 최대 동료 평가 점수
    max_coworker = 0
    # 인센티브 가능 여부
    check = [True]*length
    
    for idx in range(length):
        # 최대 동료 평가와 점수가 같거나 크다면?
        # 근무태도가 본인 점수보다 높은 사람 중 동료평가 점수까지 더 높은 사람은 없음 -> 인센티브 가능
        if max_coworker <= scores[idx][1]:
            # 최대 동료 평가 교체
            max_coworker = scores[idx][1]
            # 합산 점수가 완호의 점수보다 높다 -> 더 높은 등수 -> answer ++
            if sum(scores[idx]) > sum(wanho):
                answer += 1
                
        # 최대 동료 평가 점수보다 적다면?
        # 근무태도와 동료평가 점수 모두 본인 점수보다 높은 사람 존재
        elif max_coworker > scores[idx][1]:
            # 인센티브 불가능 표기
            check[idx] = False
            # 점수 구성이 완호와 같다면 완호도 인센티브 불가능 -> -1 리턴
            if scores[idx] == wanho:
                return -1
    
    return answer