[BOJ#2668] 숫자 고르기

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

2668번: 숫자고르기

1 초

128 MB

18706

8279

6396

45.288%

문제

세로 두 줄, 가로로 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)을 나타내는 정수 하나가 주어진다. 그 다음 줄부터는 표의 둘째 줄에 들어가는 정수들이 순서대로 한 줄에 하나씩 입력된다.

출력

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

---

제출 코드

def dfs(cur, k):
    visited[cur] = True
    
    for i in lst[cur]:
        if not visited[i]:
            dfs(i, k)
            
        elif visited[i] and k == i:
            ans.append(i)
        

N = int(input())
lst = []
for _ in range(N):
    lst.append([int(input())])
lst = [0] + lst

ans = []
for k in range(1, N+1):
    if k == lst[k]:
        ans.append(k)

for k in range(1, N+1):
    visited = [False] * (N+1)
    dfs(k, k)

ans.sort()
print(len(ans))
for j in range(len(ans)):
    print(ans[j])

싸이클을 어떻게 확인할까.. 깊은 고민을 하다 결국 합류해버린 문제

1. 자료 구조를 만들고 인덱스 값과 인덱스자리의 숫자와 같으면 결과 리스트에 미리추가, 

2. 그 후 인덱스 1번부터 1번값을 들고 dfs로 돌아주면서 들린 곳에 방문처리를 해준다.

3. 만약 첫 방문이면 계속 돌고, 방문했던 곳이면서 들고왔던 1번값과 같은 곳이면(싸이클) 숫자를 결과리스트에 추가해준다. (elif 걸린경우는 for문의 마지막 부분이다 == 싸이클확인)

 

마지막에 정렬후 출력.