[CG]2022ココア通報結果を得る


質問する

me

def solution(id_list, report, k):
    reportees={}
    reporters={}
    answer=[]
    for re in report:
        user,reported=re.split(" ")
        if reported not in reportees:
            reportees[reported]={user}
        else:
            reportees[reported].add(user)
    for reportee in reportees:
        if len(reportees[reportee])>=k:
            for r in reportees[reportee]:
                if r not in reporters:
                    reporters[r]=1
                else: reporters[r]+=1
    for id in id_list:
        if id not in reporters: answer.append(0)
        else: answer.append(reporters[id])
    return answer

others

def solution(id_list, report, k):
    answer = [0] * len(id_list)    
    reports = {x : 0 for x in id_list}

    for r in set(report):
        reports[r.split()[1]] += 1

    for r in set(report):
        if reports[r.split()[1]] >= k:
            answer[id_list.index(r.split()[0])] += 1

    return answer