백준 2668 (숫자고르기)

문제

---

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

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

 

 

 

 

 

 

풀이

---

사이클이 있는지를 확인하는 문제.

해당 숫자가 사이클이라면 list에 추가해주면 된다.

 

1	2	3	4	5	6	7
3	1	1	5	5	4	6

 

리스트에 1, 3, 5 가 추가되는 과정을 보여주면

dfs(1) -> dfs(list[1] -> 3) -> dfs(1) / 사이클 성립 O

answer.add(1)

 

dfs(2)-> dfs(1) -> dfs(3) / 사이클 성립 X

 

dfs(3) -> dfs(1) -> dfs(3) / 사이클 성립 O

answer.add(3)

 

dfs(4) -> dfs(5) -> dfs(5) / 사이클 성립 X

 

dfs(5) -> dfs(5) / 사이클 성립 O

answer.add(5)

 

dfs(6) -> dfs(4) -> dfs(5) -> dfs(5) / 사이클 성립 X

 

dfs(7) -> dfs(6) -> dfs(4) -> dfs(5) -> dfs(5) /사이클 성립 X

 

이와 같은 방식으로 DFS(깊이우선탐색)을 한다.

 

 

<전체코드>

import java.io.*;
import java.util.*;

public class Main {	
	static StringBuilder sb = new StringBuilder();
	
	
	static int n;
	static boolean[] visited;
	static int last;
	static List<Integer> answer = new ArrayList<Integer>();
	
	public static void main(String[] args) throws IOException{
		
		BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
		
		n = Integer.parseInt(bufferedReader.readLine());
		
		int[] list = new int[n + 1];
		visited = new boolean[n + 1];
		for(int i = 1; i <= n ; i++) {
			list[i] = Integer.parseInt(bufferedReader.readLine());
		}
		
		for(int i = 1 ; i <= n; i++) {
			visited[i] = true;
			last = i;
			dfs(list, list[i], visited, last);
			visited[i] = false;
		}
		
		Collections.sort(answer);
		
		sb.append(answer.size() + "\n");
		for(int n : answer) {
			sb.append(n + "\n");
		}
		
		System.out.println(sb);
	}
	
	static void dfs(int[] list, int idx, boolean[] visited, int last) {
		
		if(!visited[idx]) {
			visited[idx] = true;
			dfs(list, list[idx], visited, last);
			visited[idx] = false;
		}
		
		if(list[idx] == last) {
			answer.add(last);
		}
	}
	
}