인사고과 - 파이썬(Python)

문제 링크

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

---

입출력 예

scores	result
[[2, 2], [1, 4], [3, 2], [3, 2], [2, 1]]	4

---

입출력 예 설명

5 번째 사원은 3 번째 또는 4 번째 사원보다 근무 태도 점수와 동료 평가 점수가 모두 낮기 때문에 인센티브를 받을 수 없습니다. 2 번째 사원, 3 번째 사원, 4 번째 사원은 두 점수의 합이 5 점으로 최고점이므로 1 등입니다. 1 등이 세 명이므로 2 등과 3 등은 없고 1 번째 사원인 완호는 두 점수의 합이 4 점으로 4 등입니다.

---

풀이 과정

• 구현 문제인데 생각할 부분이 있던 문제
• 인센티브 여부를 구하는 과정이 다소 헷갈린다.
  • a와 b 각각 내림차순으로 정렬
  • a가 가장 큰 경우는, 낮아질일이 없음 -> 인센티브
  • 그 이외에 대해 b를 비교
  • a가 큰 경우에서 max_b를 누적하여 b가 max_b보다 크거나 같다면 인센티브

정답

from collections import defaultdict

def solution(scores):
    answer = 0
    
    wanho = scores[0]
    
    # 인센티브 여부 체크
    incentive = []
    scores.sort(key = lambda x : (-x[0], -x[1]))
    
    t_a = scores[0][0]
    
    # a에 대해 b를 매핑
    db_1 = defaultdict(list)  
    for a, b in scores:
        db_1[a].append(b)
    
    # 인센티브 가능한 a+b에 대해 [a, b]를 append
    db_2  = defaultdict(list)

    # a가 같은 상황에서 비교
    max_b = -1
    for a in sorted(db_1.keys(), reverse = True):
        
        sub_max_b = -1
        for b in db_1[a]:
            # a가 제일 큰 그룹에서는 인센티브 무조건 받는다.
            if a == t_a:
                db_2[a+b].append([a, b])
                max_b = max(max_b, b)
            # max_b와 비교해서 b가 크거나 같으면 인센티브 받는다.
            else:
                if b >= max_b:
                    db_2[a+b].append([a, b])
                    sub_max_b = max(sub_max_b, b)
        
            # 완호가 인센티브 못받으면 -1 리턴
            if wanho == [a, b] and not wanho in db_2[a+b]:
                return -1
            
        max_b = max(max_b, sub_max_b)
        
    # 답 계산 후, 출력
    for key in sorted(db_2.keys(), reverse=True):
        if wanho in db_2[key]:
            answer += 1
            return answer
        else:
            answer += len(db_2[key])