프로그래머스
2022 KAKAO BLIND RECRUITMENT
신고 결과 받기 문제
def solution(id_list, report, k):
answer = []
id_obj={}
stop_user=[]
report1=set(report)
report2=list(report1)
for l in id_list:
id_obj[l]={}
id_obj[l]['reported']=0
id_obj[l]['report_list']=[]
id_obj[l]['answer']=0
for i in report2:
tmp_rp=i.split(' ')
id_obj[tmp_rp[0]]['report_list'].append(tmp_rp[1])
id_obj[tmp_rp[1]]['reported'] += 1
for x in id_list:
if id_obj[x]['reported'] >= k:
stop_user.append(x)
for y in id_list:
for z in stop_user:
if z in id_obj[y]['report_list']:
id_obj[y]['answer']+=1
answer.append(id_obj[y]['answer'])
return answer
for문을 되도록이면 더 줄여보자.