2668번: 숫자고르기
세로 두 줄, 가로로 N개의 칸으로 이루어진 표가 있다. 첫째 줄의 각 칸에는 정수 1, 2, …, N이 차례대로 들어 있고 둘째 줄의 각 칸에는 1이상 N이하인 정수가 들어 있다. 첫째 줄에서 숫자를 적절
www.acmicpc.net
문제
세로 두 줄, 가로로 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
풀이 .
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
public class Main {
static int[] map = null;
static int[] check = null;
static ArrayList<Integer> cycleNodeList = null;
static boolean isCycle;
static int cycleStartNode;
public static void dfs(int node) {
int next = map[node];
if(check[next] == 0) {
check[next] = check[node];
dfs(next);
}else if(check[next] == check[node]) {
isCycle = true;
cycleStartNode = next;
}
if(isCycle) {
cycleNodeList.add(node);
}
if(node == cycleStartNode) {
isCycle = false;
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
map = new int[n + 1];
check = new int[n + 1];
cycleNodeList = new ArrayList<>();
for(int i = 1; i <= n; i++) {
map[i] = Integer.parseInt(br.readLine());
}
int dfsCnt = 1;
for(int i = 1; i <= n; i++) {
if(check[i] == 0) {
check[i] = dfsCnt;
dfs(i);
dfsCnt += 1;
}
}
Collections.sort(cycleNodeList);
System.out.println(cycleNodeList.size());
for(int node : cycleNodeList) {
System.out.println(node);
}
}
}
N제한을 제외하고는 codeung.tistory.com/177 와 완전히 동일한 문제.
여기서는 N제한이 작기 때문에 위 문제에서 시간초과 맞은 코드로도 통과할 수 있다.
(돌려보니까 통과는 하는데.. 이게 더 빠르게 돌아가네..?)
'알고리즘 문제 > 백준 온라인 저지' 카테고리의 다른 글
| [BOJ] 7576 - 토마토 JAVA (0) | 2021.01.19 |
|---|---|
| [BOJ] 4963 - 섬의 개수 JAVA (0) | 2021.01.19 |
| [BOJ] 2667 - 단지번호붙이기 JAVA (0) | 2021.01.19 |
| [BOJ] 9466 - 텀 프로젝트 JAVA (0) | 2021.01.19 |
| [BOJ] 2331 - 반복수열 JAVA (0) | 2021.01.19 |
