11-InSange

[InSange]

🔗 문제 링크

숫자 고르기

✔️ 소요된 시간

하루

✨ 수도 코드

첫번째 풀이

문제를 보았을 때 바로 코드가 번쩍였고 10분만에 풀었다.
첫째 줄과 두번째 줄을 비교했을 때 서로 방문했던 적이 있는 숫자면되고 같은 쌍이면 된다?
그러면 i가 순서대로 올라갈 때 첫번째 줄과 두번째 줄에 각자 해당하는 index번호별로 방문해줬다는 것을 표시해주면 되지 않을까?

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int N, num;
vector<bool> v1;
vector<bool> v2;
vector<int> answer;

void Solve()
{
	cin >> N;

	N += 1;
	v1.assign(N, false);
	v2.assign(N, false);

	for (int i = 1; i < N; i++)
	{
		cin >> num;
		v1[i] = true;
		v2[num] = true;

		if ((v1[i] == v2[i]) && (v2[num] == v1[num]))
		{
			answer.push_back(i);
			answer.push_back(num);
		}
	}

	sort(answer.begin(), answer.end());
	answer.erase(unique(answer.begin(), answer.end()), answer.end());

	int asize = answer.size();
	cout << asize << "\n";

	for (int i = 0; i < asize; i++)
	{
		cout << answer[i] << "\n";
	}
}

int main()
{
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	Solve();

	return 0;
}

딱 테스트 케이스를 넣자 통과되었고 문제 난이도도 골드5길래 그러려니 했더니 뙇!!

틀렸습니다..

질문게시판에 있는 여러 테스트케이스를 거치고 문제를 다시 읽어보니 내가 너무 단순하게 생각했었다..

첫번째 줄과 두번째 줄에서 이루는 집합의 원소는 서로 방문을 했을 때 그 번호까지 방문을 했다는 점
해당 부분을 체크하기 위해서는 첫번째 줄의 요소 인덱스에서 두번째 줄을 방문했을 때 결국 첫번째 줄 요소는 다시 자기 자신으로 돌아와야 된다는 점..!

두번째 풀이

그럼 결국 내가 방문했던 길을 알아야 되는 건데...
이 부분은 DFS로 모든 경로를 탐색한다고 치자.
그럼 무한 루프 4->1 1->4가 발생할 수 있으니 visited로 방문했던 경로들을 다 체크해주면 된다.
여기서 또 깊게 생각했던 부분이
순차적으로 i를 증가해서 첫번째 줄과 두번째 줄을 방문한다하면 이전의 기록들을 굳이 false로 해제할 필요가 있을까? 였다.

이 부분에 대해서 모든 테스트케이스를 긁어보기도 했고 생각도 해봤지만 아쉽게도 발견하지는 못했다.
그렇게 시도 했을 때는 역시

그럼 결국 방문했던 것들을 다시 false로 바꿔줘야하였고 그렇게 해서 겨우 통과할 수 있었다..!

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int N, num;
vector<bool> visited;
vector<int> v;
vector<int> answer;

void DFS(int cur, int start)
{
	if (visited[cur] == true)
	{
		if (start == cur) answer.push_back(start);
		return;
	}

	visited[cur] = true;
	DFS(v[cur], start);
	visited[cur] = false;
}

void Solve()
{
	cin >> N;

	N += 1;
	visited.assign(N, false);
	v.assign(N, 0);

	for (int i = 1; i < N; i++)
	{
		cin >> v[i];
	}

	for (int i = 1; i < N; i++)
	{
		DFS(i, i);
	}

	int asize = answer.size();
	cout << asize << "\n";

	for (int i = 0; i < asize; i++)
	{
		cout << answer[i] << "\n";
	}
}

int main()
{
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	Solve();

	return 0;
}

###후기
첫번째 풀이만 고집하다가 3시간이 흐르게 되었고
해당 풀이 방식으로 생각을 했을 때 바로 DFS가 떠오르지는 않았다.
블로그 글의 도움으로 아 DFS문제라는 것을 깨닫게 되었고 바로 DFS를 접목해서 이해를 하는 식으로 접근하게 되었다.

오래 걸릴만한 문제는 아니였지만 문제를 이해하는 부분에서 오점이 많았다는 것을 발견할 수 있었다..

📚 새롭게 알게된 내용

참고 블로그

[yuyu0830]

코드 깔끔하게 작성하는건 항상 볼 때마다 신기하네요. 배워갈 점이 많은 것 같습니다
DFS면 재귀로 구현이 먼저 떠오르긴 하지만 조건을 잘 잡으면 단순 while 반복문으로 처리할 수도 있습니다! 취향 차이일까요?

#include <iostream>
#include <queue>
#include <vector>

using namespace std;

typedef pair<int, int> ii;

int n, arr[102];
bool visited[102] = {0, };
bool tmp[102] = {0, };

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) 
        cin >> arr[i];

    priority_queue<int, vector<int>, greater<int> > q;

    for (int i = 1; i <= n; i++) {
        if (visited[i]) continue;

        for (int i = 1; i <= n; i++) tmp[i] = 0;

        int t = arr[i];

        vector<int> v;
        v.push_back(i);

        while (true) {
            if (t == i) {
                for (int p : v) {
                    visited[p] = 1;
                    q.push(p);
                }
                break;
            }

            if (visited[t] || tmp[t]) {
                break;
            }

            tmp[t] = true;
            v.push_back(t);
            t = arr[t];
        }
    }

    printf("%d\n", q.size());
    while (!q.empty()) {
        printf("%d\n", q.top());
        q.pop();
    }
}

[seongwon030]

방문여부를 다시 false로 초기화하고 dfs를 진행해야 사이클이 존재하는지 알 수 있는거네요. 저는 dfs 함수를 따로 만들어 재귀적으로 푸는게 좀 어려워서 while에 넣고 스택이 비워질 때 까지 반복했어요.

for i in range(1, n+1):
    visited = [0] * (n+1)
    stack = [i]
    visited[i] = 1

    while stack:
        v = stack.pop()
        for j in graph[v]:
            if not visited[j]:
                stack.append(j)
                visited[j] = 1
            elif visited[j] and i == j: 
                result.add(j)

[dhlee777]

다른 경우의 수들을 위해 탐색후 visited=false 처리를 해주어야 하는군요 visited.assign(N, false); v.assign(N, 0); 이런방식으로 벡터를초기화 하는법도 잘 보았습니다. 저도 이런 깔끔한 코드를 짜는날까지 노력하겠습니다..

[InSange]

코드 깔끔하게 작성하는건 항상 볼 때마다 신기하네요. 배워갈 점이 많은 것 같습니다 DFS면 재귀로 구현이 먼저 떠오르긴 하지만 조건을 잘 잡으면 단순 while 반복문으로 처리할 수도 있습니다! 취향 차이일까요?

#include <iostream>
#include <queue>
#include <vector>

using namespace std;

typedef pair<int, int> ii;

int n, arr[102];
bool visited[102] = {0, };
bool tmp[102] = {0, };

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) 
        cin >> arr[i];

    priority_queue<int, vector<int>, greater<int> > q;

    for (int i = 1; i <= n; i++) {
        if (visited[i]) continue;

        for (int i = 1; i <= n; i++) tmp[i] = 0;

        int t = arr[i];

        vector<int> v;
        v.push_back(i);

        while (true) {
            if (t == i) {
                for (int p : v) {
                    visited[p] = 1;
                    q.push(p);
                }
                break;
            }

            if (visited[t] || tmp[t]) {
                break;
            }

            tmp[t] = true;
            v.push_back(t);
            t = arr[t];
        }
    }

    printf("%d\n", q.size());
    while (!q.empty()) {
        printf("%d\n", q.top());
        q.pop();
    }
}

문제를 이해하는 부분에 대해서 그런것 같습니다!! 리뷰에 써져있듯 저는 문제 방향성 자체가 정답을 유추하는 것과는 거리가 멀어서.. 처음에 dfs로 가능한 경로의 위치들을 파악하고 진행을 하다보니 while문에 대해서 고려는 해보지 못했습니다.

while문에서 제약 조건들을 추가하여 풀이하는 방법 잘 봤습니다! 하지만 저에게는 이해하는 시간을 비롯해 조금 어렵더라구요... 더군다나 dfs에 물들어져 있다보니 사고전환이 필요할 것 같습니다.

[InSange]

방문여부를 다시 false로 초기화하고 dfs를 진행해야 사이클이 존재하는지 알 수 있는거네요. 저는 dfs 함수를 따로 만들어 재귀적으로 푸는게 좀 어려워서 while에 넣고 스택이 비워질 때 까지 반복했어요.

for i in range(1, n+1):
    visited = [0] * (n+1)
    stack = [i]
    visited[i] = 1

    while stack:
        v = stack.pop()
        for j in graph[v]:
            if not visited[j]:
                stack.append(j)
                visited[j] = 1
            elif visited[j] and i == j: 
                result.add(j)

매번 초기화 한다는 것은 비용이 많이 들 것 같은데.. 이에 따른 비교같은 것도 있으면 좋을 것 같아요!!

[InSange]

11-InSange