[파이썬] [DFS] 백준 2668 숫자고르기

문제

https://www.acmicpc.net/problem/2668

2668번: 숫자고르기

코드

n=int(input())
graph=[[] for _ in range(n+1)]
for i in range(1,n+1):
    t=int(input())
    graph[i].append(t)

answer=set()

def dfs(first,second,num):
    #문제 그림의 첫번째 줄
    first.add(num)
    #문제 그림의 두번째 줄
    second.add(graph[num][0])

    if graph[num][0] in first:
        #첫번째 줄 집합과 두번째 줄 집합이 같다면
        if first==second:
            answer.update(first)
            return True
        else:
            return False
    else:
        dfs(first,second,graph[num][0])

for i in range(1,n+1):
    dfs(set(),set(),i)

print(len(answer))
print(*sorted(list(answer)),sep="\n")