[프로그래머스] 가장 많이 받은 선물

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

프로그래머스

---

문제 요약

- 친구들이 이번 달 선물을 주고받은 기록을 바탕으로 다음 달 누가 선물을 많이 받을지 예측

- 규칙

 1. 두 사람 선물 주고 받은 기록이 있다면 ? 더 선물을 많이 준 사람이 다음 달에 선물을 하나 받음

 2. 두 사람 선물 주고 받은 기록이 없거나 같다면 ? 선물 지수가 더 큰 사람이 선물을 하나 받음 (이 때, 선물지수도 같다면 주고 받지 않음)

 3. 선물지수 = (이번 달까지 자신이 친구들에게 준 선물 수) - (이번 달까지 자신이 친구들에게 받은 선물 수)

- friends는 친구들의 이름이 담긴 배열, gifts는 주고받은 선물 기록을 담은 배열, answer은 가장 많은 선물을 받는 친구가 받을 선물의 수를 의미함

코드

def solution(friends, gifts):
    answer = 0
    
    # 주고받은 선물 
    table = [[0] * len(friends) for _ in range(len(friends))]
    for name in gifts:
        give, receive = name.split(' ')
        table[friends.index(give)][friends.index(receive)] += 1

    # 선물 지수
    give_idx = [0] * len(friends)
    receive_idx = [0] * len(friends)
    for i in range(len(friends)):
        for j in range(len(friends)):
            give_idx[i] += table[i][j]
            receive_idx[i] += table[j][i]
    gift_idx = [give_idx[i] - receive_idx[i] for i in range(len(friends))]

    # 선물을 가장 많이 받을 친구의 선물 수 계산
    for i in range(len(friends)):
        tmp = 0
        for j in range(len(friends)):
            if table[i][j] > table[j][i]:
                tmp += 1
            elif table[i][j] == table[j][i]:
                if gift_idx[i] > gift_idx[j]:
                    tmp += 1
        if tmp > answer:
            answer = tmp

    return answer

 

 문제를 풀기 전 문제에서 주어진 입출력 예를 찬찬히 살펴보면 더욱 쉽게 풀 수 있다.

 우선 1번째 표인 주고 받은 선물의 수를 나타낸 2차원 배열 table을 생성하였다.index 함수를 이용하여 friends 배열에서의 인덱스를 알아내어 table에 주고받은 선물을 저장하였다.

 2번째 표인 선물 지수는 준 선물을 give_idx, 받은 선물을 receive_idx에 저장한 후 두 수를 뺀 선물 지수를 gift_idx에 저장하였다. 다른 코드를 참고한 결과 선물 지수 코드는 아래와 같이 작성할 수도 있었고, 코드가 더욱 간결해졌다. 2차원 배열에서 행의 합과 열의 합을 계산할 때 유용하게 사용될 것 같으니 꼭 기억해두자.

gift_idx = [0] * len(friends)
for i in range(len(friends)):
	gift_idx[i] = sum(table[i]) - sum([k[i] for k in table])

 

 이제 주어진 입출력을 이용하여 선물을 가장 많이 받을 친구의 선물 수를 계산하면 된다. 각 사람의 받을 선물 수를 계산한 tmp와 answer를 비교하여 tmp를 갱신해주면 된다.