[Python/파이썬] 백준 2668번: 숫자고르기

문제

 

풀이

사이클을 찾는 문제이다.

아래 풀이와 거의 동일하다.

https://ku-hug.tistory.com/169

[Python] 백준 9466번: 텀 프로젝트

 

코드

import sys
input = sys.stdin.readline

n = int(input())
graph = [0]
visited = [False] * (n + 1)
cycle = []
ans = []

for _ in range(n):
    graph.append(int(input()))
    
def dfs(x):
    visited[x] = True
    cycle.append(x)
    
    next = graph[x]
    if visited[next]:
        if next in cycle:
            for c in cycle[cycle.index(next):]:
                ans.append(c)
        return
    dfs(next)

for i in range(1, n + 1):
    if not visited[i]:
        cycle = []
        dfs(i)

print(len(ans))
ans.sort()
for a in ans:
    print(a)

 

마무리

사이클 문제를 한번 파악해서 쉽게 풀 수 있었다.