백준 온라인 저지, DFS / 2668번: 숫자고르기 (파이썬 / 백준 골드문제)

문제

세로 두 줄, 가로로 N개의 칸으로 이루어진 표가 있다. 첫째 줄의 각 칸에는 정수 1, 2, …, N이 차례대로 들어 있고 둘째 줄의 각 칸에는 1이상 N이하인 정수가 들어 있다. 첫째 줄에서 숫자를 적절히 뽑으면, 그 뽑힌 정수들이 이루는 집합과, 뽑힌 정수들의 바로 밑의 둘째 줄에 들어있는 정수들이 이루는 집합이 일치한다. 이러한 조건을 만족시키도록 정수들을 뽑되, 최대로 많이 뽑는 방법을 찾는 프로그램을 작성하시오. 예를 들어, N=7인 경우 아래와 같이 표가 주어졌다고 하자.

이 경우에는 첫째 줄에서 1, 3, 5를 뽑는 것이 답이다. 첫째 줄의 1, 3, 5밑에는 각각 3, 1, 5가 있으며 두 집합은 일치한다. 이때 집합의 크기는 3이다. 만약 첫째 줄에서 1과 3을 뽑으면, 이들 바로 밑에는 정수 3과 1이 있으므로 두 집합이 일치한다. 그러나, 이 경우에 뽑힌 정수의 개수는 최대가 아니므로 답이 될 수 없다.

입력

첫째 줄에는 N(1≤N≤100)을 나타내는 정수 하나가 주어진다. 그 다음 줄부터는 표의 둘째 줄에 들어가는 정수들이 순서대로 한 줄에 하나씩 입력된다.

출력

첫째 줄에 뽑힌 정수들의 개수를 출력하고, 그 다음 줄부터는 뽑힌 정수들을 작은 수부터 큰 수의 순서로 한 줄에 하나씩 출력한다.

제한

예제 입력 1

7
3
1
1
5
5
4
6

예제 출력 1

3
1
3
5

힌트

접근 방법

- 리스트의 인덱스를 하나씩 탐색하며 그 인덱스의 값을 기준으로 dfs 탐색을 진행한다.
- 주의 사항
- - 리스트를 하나씩 탐색할 때마다 방문처리 리스트를 매번 초기화한다.
- - dfs를 통해 처음 위치로 다시 방문을 하게된다면 방문처리한 개수를 모두 저장한다.
- - 집합에 포함된 인덱스에 대해선 따로 방문처리를 진행하여 나중에 방문처리를 할 수 없도록 한다.

코드

# https://www.acmicpc.net/problem/2668
# 접근방법
# 리스트의 인덱스를 하나씩 탐색하며 그 인덱스의 값을 기준으로 dfs 탐색을 진행한다.
# 주의 사항
# - 리스트를 하나씩 탐색할 때마다 방문처리 리스트를 매번 초기화한다.
# - dfs를 통해 처음 위치로 다시 방문을 하게된다면 방문처리한 개수를 모두 저장한다.
# - 집합에 포함된 인덱스에 대해선 따로 방문처리를 진행하여 나중에 방문처리를 할 수 없도록 한다.
def dfs(s, x):
    visited[x] = True
    if visited_set[x]:
        return False
    if arr[x] == s:
        return True
    if visited[arr[x]]:
        return False

    if dfs(s, arr[x]):
        return True
    return False


n = int(input())
arr = [0] + list(int(input()) for _ in range(n))
visited_set = [False for _ in range(n+1)]
for i in range(1, n+1):
    visited = [False for _ in range(n+1)]
    if dfs(i, i):
        for j in range(1, n+1):
            if visited[j]:
                visited_set[j] = True
print(sum([1 if x else 0 for x in visited_set ]))
for i in range(1, n+1):
    if visited_set[i]:
        print(i)