boj2668 숫자 고르기

링크

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

출력

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

나의 풀이

• 이 블로그 의 풀이를 참고하였다.
• 이 문제는 사이클이 생기는 것을 찾아야 한다. 시작하는 것과 끝이 동일하면 사이클이 생기는 것이므로, answer에 추가해준다.
• 핵심 코드

  private static void dfs(int start,int now){
          if(!checked[nums[now]]){
              checked[nums[now]] = true;
              dfs(start,nums[now]);
              checked[nums[now]] = false;
          }
          if(nums[now]==start) answer.add(now);
      }

• 전체 코드

  package Baekjoon.java.gold;
  
  import java.io.BufferedReader;
  import java.io.InputStreamReader;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.stream.Collectors;
  
  public class boj2668 {
      static int nums[],N,maxCnt;
      static boolean[] checked;
      static List<Integer> answer;
      public static void main(String[] args) throws Exception{
          BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
          N = Integer.parseInt(br.readLine());
          nums = new int[N+1];
  
          maxCnt=0;
          for (int i = 1; i < N+1; i++) {
              nums[i] = Integer.parseInt(br.readLine());
          }
          checked = new boolean[N+1];
          answer = new ArrayList<Integer>();
          for (int i = 1; i<=N; i++) {
              checked[i] = true;
              dfs(i,i);
              checked[i] = false;
          }
          answer = answer.stream().sorted().distinct().collect(Collectors.toList());
          System.out.println(answer.size());
          for (int i = 0; i < answer.size(); i++) {
              System.out.println(answer.get(i));
          };
      }
  
      private static void dfs(int start,int now){
          if(!checked[nums[now]]){
              checked[nums[now]] = true;
              dfs(start,nums[now]);
              checked[nums[now]] = false;
          }
          if(nums[now]==start) answer.add(now);
      }
  }

결과