[백준] 2668번_숫자고르기

문제 링크 : https://www.acmicpc.net/problem/2668

2668번: 숫자고르기

n = int(input())
dic = {}
for i in range(n):
    dic[i+1] = int(input())

while True:
    baseSet = set(dic.values())
    dic = {key:value for key, value in dic.items() if key in baseSet}
    if baseSet == set(dic.values()):
        break
print(len(dic))
for key in dic.keys():
    print(key)

딕셔너리를 이용해 풀어봤다.