2668. 숫자 고르기 GOLD V(C++)

https://www.acmicpc.net/problem/2668

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

출력

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

 

 

 

#include <iostream>
#include <map>
#include <queue>
#include <vector>
 
using namespace std;
 
int n;
bool visit[101];
map<int, int> m;
priority_queue<int, vector<int>, greater<int> > pq;
 
bool travel(int idx)
{
    int start = idx;
    for(int i=0;i<n;i++)
    {
        idx = m[idx];
        if(idx == start) return true;
        if(visit[idx]) return false;
    }
 
    return false;
}
 
void solve()
{
    for(int i=1;i<=n;i++)
    {
        if(i == m[i] && !visit[i])
        {
            pq.push(i);
            visit[i] = true;
        }
        else{
            if(travel(i))
            {
                int temp = i;
                while(true)
                {
                    visit[temp] = true;
                    temp = m[temp];
                    pq.push(temp);
                    if(temp == i) break;
                }
            }
        }
    }
 
}
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n;
 
    for(int i=1;i<=n;i++)
        cin>>m[i];
 
    solve();
    cout<<pq.size()<<'\n';
    while(!pq.empty())
    {
        cout<<pq.top()<<'\n';
        pq.pop();
    }
    return 0;
}
 

고리 생성을 찾아야 한다. 

예제는 2개가 고리를 이루는 것만 보여줬는데. 최대 n만큼의 고리가 생성될 수 있다. 

출력은 우선순위큐를 이용했는데 배열이나 백터로 해도 상관 없다.

 

MAP 라이브러리를 이용했다. key와 value를 이용하기 위해서 사용했다.

 

1. 만약 key 와 value가 동일하다면 그 부분 true로 바꾸어 주고 우선순위 큐에 push

2. 만약 다르다면 그 위치에서 여행을 떠난다.

고리가 생성되는것을 찾아야 한다. 아무리 길어도 고리가 n 보다는 크지 않을 것이므로 최대 n 번만 돌아준다.

3. 고리가 생성되면 true를 리턴하고 크기 안에 고리를 생성하지 못할 때는 false를 리턴

4. true를 리턴 받았을 때는 똑같은 방식으로 우선순위 큐에 값들을 집어 넣어준다.