from collections import defaultdict
def solution(id_list, report, k):
user_information = defaultdict(list)
result = [0] * len(id_list)
for report_inform in report:
tmp_list = report_inform.split(' ')
if tmp_list[0] not in user_information[tmp_list[1]]:
user_information[tmp_list[1]].append(tmp_list[0])
for i in id_list:
if len(user_information[i]) >= k:
for value in user_information[i]:
result[id_list.index(value)] += 1
return result
defaultdict를 이용했다.
딕셔너리에서 값인 리스트에 원소를 추가하기 위해 사용하였다.
from collections import defaultdict # import
user_information = defaultdict(list) # list를 값으로 하는 dict 생성
user_information['키값'].append('넣을 값') # 키 값에 해당하는 곳 리스트에 값을 추가