BOJ Q2668 [ 골드5 ]
문제
세로 두 줄, 가로로 N개의 칸으로 이루어진 표가 있다. 첫째 줄의 각 칸에는 정수 1, 2, …, N이 차례대로 들어 있고 둘째 줄의 각 칸에는 1이상 N이하인 정수가 들어 있다. 첫째 줄에서 숫자를 적절히 뽑으면, 그 뽑힌 정수들이 이루는 집합과, 뽑힌 정수들의 바로 밑의 둘째 줄에 들어있는 정수들이 이루는 집합이 일치한다. 이러한 조건을 만족시키도록 정수들을 뽑되, 최대로 많이 뽑는 방법을 찾는 프로그램을 작성하시오. 예를 들어, N=7인 경우 아래와 같이 표가 주어졌다고 하자.
| 1 | 2 | 3 | 4 | 5 | 6 | 7 |
|---|---|---|---|---|---|---|
| 3 | 1 | 1 | 5 | 5 | 4 | 6 |
이 경우에는 첫째 줄에서 1, 3, 5를 뽑는 것이 답이다. 첫째 줄의 1, 3, 5밑에는 각각 3, 1, 5가 있으며 두 집합은 일치한다. 이때 집합의 크기는 3이다. 만약 첫째 줄에서 1과 3을 뽑으면, 이들 바로 밑에는 정수 3과 1이 있으므로 두 집합이 일치한다. 그러나, 이 경우에 뽑힌 정수의 개수는 최대가 아니므로 답이 될 수 없다.
입력
첫째 줄에는 N(1≤N≤100)을 나타내는 정수 하나가 주어진다. 그 다음 줄부터는 표의 둘째 줄에 들어가는 정수들이 순서대로 한 줄에 하나씩 입력된다.
출력
첫째 줄에 뽑힌 정수들의 개수를 출력하고, 그 다음 줄부터는 뽑힌 정수들을 작은 수부터 큰 수의 순서로 한 줄에 하나씩 출력한다.
예제
입력
7
3
1
1
5
5
4
6출력
3
1
3
5풀이
N = int(input()) # 입력
arr = [0] # 두번째 줄 숫자 담을 리스트.
for _ in range(N):
arr.append(int(input()))
answer = set() # 결과 담을 set
# dfs 정의
def dfs(first, second, num):
first.add(num) # 첫번째 줄 집합에 num 추가
second.add(arr[num]) # 두번째 줄 집합에 arr[num] 추가
if arr[num] in first: # arr[num]이 첫번째 줄 집합에 있을 때
if first == second: # 첫번째 줄 집합과 두번째 줄 집합이 같다면
answer.update(first) # 결과 업데이트
return True
return False
return dfs(first, second, arr[num]) # 아니라면 다음 dfs 실행
# dfs 실행
for i in range(1, N+1):
if i not in answer: # i가 결과 집합 안에 없는 경우만 실행
dfs(set(), set(), i)
print(len(answer))
print(*sorted(list(answer)), sep='\n')골드5 치고는 너무 쉬운 문제
dfs에서 arr[num]이 첫번째 줄 집합에 있고 첫번째 줄 집합과 두번째 줄 집합이 다르다면 return False를 하여 dfs를 끝낸다. 이 때 끝내지 않으면 오류가 발생할 수 있다.
예를 들어 예제의 숫자에서 i=2일 때 dfs를 시작하게 되면, 2->1, 1->3, 3->1이 되서 첫번째 줄 집합은 {1, 2, 3}, 두번째 줄 집합은 {1, 3} 인 상태에서 두 집합은 다르므로 계속 dfs를 실행하고, recursion error가 생긴다.
