[Gold V] 숫자고르기 - 2668 (JAVA)

문제 요약

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

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

---

문제 분석

문제를 봐도 도저히 이해가 안가서 찾아본 결과, 사이클이 가능한 수를 찾으면 된다는 것을 알았다.

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Main {
    static int[] arr;
    static int N;
    static List<Integer> cycle;

    /*
        첫째 줄에서 숫자를 적절히 뽑으면, 그 뽑힌 정수들이 이루는 집합과, 뽑힌 정수들의 바로 밑의 둘째 줄에 들어있는 정수들이 이루는 집합이 일치한다.
        이 부분이 이해 안돼서 검색해봄
        검색 후 이게 사이클을 의미한다는것을 알게됨
        모든 수를 탐색하며 사이클인 수를 찾아 출력해준다.
     */
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        // 1 ~ N까지기때문에 N+1로 설정한다.
        arr = new int[N+1];
        for(int i = 1; i <= N; i++){
            arr[i] = Integer.parseInt(br.readLine());
        }
        // 반복되는 수를 넣기 위한 리스트
        cycle = new ArrayList<>();
        // 반복문을 통해 모든 수에 대한 사이클 찾기
        for(int i = 1; i <= N; i++){
            // 해당 수로 시작해서 해당 수로 끝나기까지의 과정을 확인하기 위한 방문배열
            boolean[] v = new boolean[N+1];
            // 시작숫자, 현재숫자(지금은 시작숫자), 방문배열을 dfs에 넣고 탐색한다.
            dfs(i, i, v);
        }
        // 작은수부터 출력하기 위한 정렬
        Collections.sort(cycle);
        System.out.println(cycle.size());
        for(int i : cycle){
            System.out.println(i);
        }
    }

    private static void dfs(int startNum, int currNum, boolean[] v) {
        // 종료조건
        // 1. 이번 노드가 방문한 노드인지 확인
        if(v[currNum] == true) {
            // 2. 사이클인지 확인하기 위해 현재수와 끝나는 수가 같은지 확인
            if (startNum == currNum) {
                cycle.add(startNum);
            }
            return;
        }
        // 현재 수를 방문했다는 체크
        v[currNum] = true;
        // currNum을 다음 수로 바꾼 후 다시 재귀
        dfs(startNum, arr[currNum], v);
    }
}

---

정리

새로 배운 내용

사이클을 찾는 문제를 한번 풀어보고싶었는데, 처음 경험해 봤다.

개선할 점

없음

더 찾아볼만한 사항

없음

---