BOJ 백준온라인져지 2668 숫자고르기 풀이

1. 사이클 체크하는 문제

2. DFS

3. 사이클 체크는 매 순간 visited 를 갱신 해줘야됨

4. 사이클이 있다면, 그것들을 출력해줘야됨 (오름차순)

	import java.util.*;
	import java.io.*;
	/**
	* https://www.acmicpc.net/problem/2668
	* BOJ 백준온라인져지 2668 숫자고르기 풀이
	*/
	public class Main {
	private static int arr[];
	private static boolean visited[];
	public static void main (String args[]) throws IOException{
	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
	int N = Integer.parseInt(br.readLine());
	arr = new int[N + 1];
	for (int i = 1; i <= N; i++) {
	arr[i] = Integer.parseInt(br.readLine());
	}
	ArrayList<Integer> list = new ArrayList<>();
	for (int i = 1; i <= N; i++) {
	visited = new boolean[N + 1];
	if (DFS(i, i)) {
	if (list.contains(i)) {
	continue;
	}
	list.add(i);
	int next = arr[i];
	while (next != i) {
	list.add(next);
	next = arr[next];
	}
	}
	}
	bw.write(list.size() + "\n");
	Collections.sort(list);
	for (int i = 0; i < list.size(); i++) {
	bw.write(list.get(i) + "\n");
	}
	bw.flush();
	}
	private static boolean DFS (int x, int startX) {
	if (visited[x]) return false;
	visited[x] = true;
	if (arr[x] == startX) {
	return true;
	} else {
	return DFS(arr[x], startX);
	}
	}
	}

view raw Main.java hosted with ❤ by GitHub

문제

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

출력

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