숫자 고르기

문제

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

출력

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

 

 

import java.util.*;

class Main {
	static int n;
	static int[] arr;	// 밑 숫자 배열
	static boolean[] visited;	// 그 전에 숫자를 뽑았는지 여부
	static int start;	// 탐색 시작 숫자
	static ArrayList<Integer> list = new ArrayList<Integer>();	// 최종 리스트

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);

		n = scan.nextInt();

		arr = new int[n + 1];
		visited = new boolean[n + 1];

		for (int i = 1; i <= n; i++) {
			arr[i] = scan.nextInt();
		}

		for (int i = 1; i <= n; i++) {	// 숫자 하나하나 전부 탐색
			visited[i] = true;	// 시작 숫자 true
			start = i;
			dfs(i);
			visited[i] = false;	// 다음 dfs를 위해 false
		}

		System.out.println(list.size());

		for (int i = 0; i < list.size(); i++) {
			System.out.println(list.get(i));
		}
	}

	static void dfs(int idx) {	// dfs
		if (!visited[arr[idx]]) {	// 뽑은적이 없다면
			visited[arr[idx]] = true;
			dfs(arr[idx]);
			visited[arr[idx]] = false;	// 다음 dfs를 위해 false
		}
		if (arr[idx] == start) {	// 사이클 마지막이 시작 숫자와 동일할 때 list에 추가
			list.add(start);
		}
	}

}

 

문제출처

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