[BOJ][백준][JAVA] 2668 - 숫자고르기

👩‍💻 문제 풀러 가기

2668번: 숫자고르기

👩‍💻 문제

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

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

👩‍💻 해결 방법

DFS를 돌리면서 재귀로 하나씩 확인해가며 visit를 체크해주면 되는 문제였다. 예를 들어, dfs(1,1)을 넣고 시작하면 위의 예시로 (3,1) 이 확인되고 3에 들어있는 값이 1이기 때문에 다시 (1,1)이 되면 동적 배열에 값을 담고 다음 (2,2)로 넘어가며 확인하는 형식이다.

👩‍💻 소스코드

package BOJ.bfs_dfs;

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;

public class boj2668_숫자고르기 {
    static int N;
    static int[] arr;
    static boolean[] visit;
    static ArrayList<Integer> result;

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());

        arr = new int[N + 1];
        visit = new boolean[N + 1];
        result = new ArrayList<>();

        for (int i = 1; i <= N; i++) {
            arr[i] = Integer.parseInt(br.readLine());
        }

        for (int j = 1; j <= N; j++) {
            visit[j] = true;
            dfs(j, j);
            visit[j] = false;
        }

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

    static void dfs(int value, int i) {
        if (visit[arr[value]] == false) {
            visit[arr[value]] = true;
            dfs(arr[value], i);
            visit[arr[value]] = false;
        }
        if (arr[value] == i) {
            result.add(arr[value]);
        }
    }
}

👩‍💻 리뷰

내 제출

---

조합 문제를 많이 풀다 보니 보자마자 감잡고 문제를 풀었던 것 같다😂😂😂