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

출력

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

예제 입력 1 복사

7 3 1 1 5 5 4 6

예제 출력 1 복사

3 1 3 5

---

기초 dfs 문제였다.

입력으로 들어온 수열에서 사이클을 이루는 것이 있는지를 판별하고 추가하면 되는 문제다.

1부터 N까지 차례대로 dfs 검사를 하며 사이클을 이루면 추가하고 아니면 넘어간다.

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/BaekJoon/2020/%EC%88%AB%EC%9E%90%EA%B3%A0%EB%A5%B4%EA%B8%B0%20(2668%EB%B2%88).cpp

chosh95/STUDY

C++ 코드

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
int N;
int p[101];
int visit[101];
vector<int> res;

void dfs(int start, int cur) {
	if (visit[cur] == 0) {
		visit[cur] = start;
		dfs(start, p[cur]);
	}
	else if (cur == start) {
		res.push_back(start);
		return;
	}
}

int main() {
	cin >> N;
	for (int i = 1; i <= N; i++)
		cin >> p[i];
	
	for (int i = 1; i <= N; i++) {
		memset(visit, 0, sizeof(visit));
		dfs(i, i);
	}

	cout << res.size() << "\n";
	for (int i = 0; i < res.size(); i++) {
		cout << res[i] << "\n";
	}
}