[백준] 2668번: 숫자고르기 (python)

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

 

 

출력 :

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

 

 

풀이)

이 문제를 정리하자면, 결국 그래프들을 살펴보면서 사이클이 몇개 있는지를 파악하는 문제이다.

따라서 다음과 같이 코드를 짜보았다.

 

def dfs(v, start):
    global cnt, ans
 
    visited[v] = 1
 
    # 다시 시작점으로 돌아왔을 경우
    if v == start:
        cnt += 1
        ans.append(start)
        return
 
    # 앞으로 갈 곳이 방문하지 않은 곳이라면
    if visited[g[v]] == 0:
        dfs(g[v], start)
 
 
# 그래프에서 사이클의 갯수 구하기
n = int(input())
# 숫자당 간선이 하나이므로 1차원 배열 활용
g = [0]
for i in range(1, n + 1):
    g.append(int(input()))
 
cnt = 0
ans = []
# 모든 숫자를 순회
for i in range(1, n + 1):
    # 방문 처리 배열 초기화
    visited = [0] * (n + 1)
    dfs(g[i], i)
 
print(cnt)
for i in ans:
    print(i)

---

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

2668번: 숫자고르기