[백준_2668번] 숫자고르기

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

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

 

[코드]

import java.util.*;

public class BaekJoon_2668 {
	static ArrayList<Integer> arr=new ArrayList<>();
	static boolean[] visited;
	static int[] num;
	static int n;
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();
		
		visited=new boolean[n+1];
		num=new int[n+1];
		
		for(int i=1;i<=n;i++)
			num[i]=sc.nextInt();
		
		// 사이클이 발생하는지 확인
		for(int i=1;i<=n;i++) {
			visited[i]=true;
			dfs(i,i);
			visited[i]=false;
		}
		
		Collections.sort(arr);
		System.out.println(arr.size());
		for(int i=0;i<arr.size();i++)
			System.out.println(arr.get(i));
	}
	
	static void dfs(int start, int target) {
		// 사이클이 만들어지는 경우
		if(num[start]==target)
			arr.add(target);
		
		if(!visited[num[start]]) {
			visited[num[start]]=true;
			dfs(num[start],target);
			visited[num[start]]=false;
		}
	}
}

 

[고찰]

 이번 문제는 그림을 그려가며 이해하다 보면 그래프의 사이클을 찾는 문제라는 것을 알 수 있었다. 그래프의 사이클은 DFS 알고리즘을 사용하여 찾아야 하기 때문에 해당 알고리즘을 사용하여 코드를 짜주면 정답 처리를 받을수 있었다.