[BAEKJOON ONLINE JUDGE] 2668번 : 숫자고르기

시간 제한 : 1 초

 

메모리 제한 : 128 MB

 

문제

세로 두 줄, 가로로 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)을 나타내는 정수 하나가 주어진다. 그 다음 줄부터는 표의 둘째 줄에 들어가는 정수들이 순서대로 한 줄에 하나씩 입력된다.

 

출력

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

---

#include<stdio.h>
#include<stdlib.h>
#include<stdbool.h>

int main() {
	int n = 0;
	scanf(" %d", &n);

	int** arr = (int**)malloc(sizeof(int*) * n);
	for (int i = 0; i < n; i++) {
		arr[i] = (int*)malloc(sizeof(int) * 2);
		arr[i][0] = i + 1;
		scanf(" %d", &arr[i][1]);
	}

	int count = 0;
	int* select_arr = NULL;
	for (int i = 0; i < n; i++) {
		int* repeat_arr = (int*)malloc(sizeof(int) * n);
		int index = 0;
		repeat_arr[index++] = arr[i][0];

		bool repeat = true;
		while (repeat) {
			if (repeat_arr[index - 1] == 0)
				break;

			int repeat_index = repeat_arr[index - 1] - 1;
			if (repeat_arr[0] == arr[repeat_index][1]) {
				for (int j = 0; j < index; j++) {
					arr[repeat_arr[j] - 1][0] = 0;
					arr[repeat_arr[j] - 1][1] = 0;
				}

				count += index;

				int* tmp_arr = (int*)malloc(sizeof(int) * count);
				for (int j = 0; j < count; j++)
					if (j < count - index)
						tmp_arr[j] = select_arr[j];
					else
						tmp_arr[j] = repeat_arr[count - j - 1];
					
				free(select_arr);
				select_arr = tmp_arr;
			}

			for (int i = 1; i < index; i++)
				if (repeat_arr[i] == arr[repeat_index][1])
					repeat = false;

			repeat_arr[index++] = arr[repeat_index][1];
		}
		free(repeat_arr);
	}

	for (int i = 0; i < count; i++)
		for (int j = i + 1; j < count; j++)
			if (select_arr[i] > select_arr[j]) {
				int tmp = select_arr[i];
				select_arr[i] = select_arr[j];
				select_arr[j] = tmp;
			}

	printf("%d\n", count);
	for (int i = 0; i < count; i++)
		printf("%d\n", select_arr[i]);

	return 0;
}