[백준 2668] 숫자 고르기

#include<cstdio>
#include<algorithm>
using namespace std;
int arr[1002];
bool visited[1002];
bool finished[1002];
int ans[102];
int i,n,temp,tot,cnt;
void func(int node){
    visited[node]=true;
    int next=arr[node];
    
    if(visited[next]){
        if(!finished[next]){
            for(i=next; i!=node; i=arr[i]){
                cnt++;
                ans[cnt]=i;
            }
            cnt++; ans[cnt]=i;
        }
    } else
        func(next);
    finished[node]=true;
}
int main(void){
    scanf("%d", &n);
    for(int i=1; i<=n; i++){
        scanf("%d", &arr[i]);
    }
    
    for(int i=1; i<=n; i++){
        if(!visited[i]){
            func(i);
        }
    }
    
    sort(ans+1,ans+cnt+1);
    printf("%d\n",cnt);
    for(int i=1; i<=cnt; i++)
        printf("%d\n", ans[i]);
    return 0;
}

9466 텀프로젝트(https://withallmy.tistory.com/186)와 매우 비슷하다.

물론 한번에 풀지는 못했...^_T