파이썬 알고리즘 : 신고 결과 받기

2차원 배열과 INDEX

2023년 11월 13일 알고리즘 문제풀이

문제

신고 결과 받기

난이도

Lv.1

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
def solution(id_list, report, k):
    n = len(id_list)
    answer = [0 for _ in range(n)]
    table = [[0 for _ in range(n)] for _ in range(n)]
    
    for x in report:
        sender, receiver = x.split()
        a = id_list.index(sender)
        b = id_list.index(receiver)
        table[b][a] = 1
        
    for i in table:
        if sum(i) >= k:
            for j in range(len(i)):
                if i[j]:
                    answer[j] += 1
            
    return answer

어떤 배열들을 만들어야 하는지 헷갈렸지만 크게 어려운 문제는 아니었다.