Passing Yearbook

def findSignatureCounts(arr):
    res = [0] * len(arr)
    arr = [x - 1 for x in arr]
    visited = set()

    for i in range(len(arr)):
        if i not in visited:
            temp_set = {i}
            visited.add(i)
            j=i
            while(True):
                print(j,arr[j])
                if arr[j] == i:
                    break
                else:
                    j = arr[j]
                    temp_set.add(j)
                    visited.add(j)
            for num in temp_set:
                res[num] = len(temp_set)
    return res