어디 부분이 잘못되었는지 잘 모르겠습니다.

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

2-S

해결된 질문

작성

반례들을 다 찾아서 넣어봐도 다 맞게 나오는데 어디 부분이 틀렸는지 알 수 있을까요?

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;

int N, M, A, B, visited[10004], cnt;
vector<int> adj[10004];
vector<pair<int, int>> v;

bool cmp(pair<int, int> a, pair<int, int> b)
{
	return a.second > b.second;
}

void dfs(int num)
{
	if (visited[num])
		return;
	visited[num] = 1;
	cnt++;

	for (int i = 0; i < adj[num].size(); i++)
	{
		dfs(adj[num][i]);
	}
}

int main()
{
	cin >> N >> M;
	for (int i = 0; i < M; i++)
	{
		cin >> A >> B;
		adj[B].push_back(A);
	}

	for (int i = 1; i <= N; i++)
	{
		memset(visited, 0, sizeof(visited));
		cnt = 0;
		dfs(i);
		v.push_back({i, cnt});
	}
	
	sort(v.begin(), v.end(), cmp);

	int ret = v[0].second;
	for (auto i : v)
	{
		if (ret > i.second)
			break;
		cout << i.first << " ";
	}

	return 0;
}

c++코딩-테스트

답변 2

큰돌

지식공유자

안녕하세요 가인님 ㅎㅎ

마지막 부분만 이런식으로 고쳐보시겠어요?

    int maxCnt = 0;

    for (int i = 1; i <= N; ++i)
    {
        memset(visited, 0, sizeof(visited));
        cnt = 0;
        dfs(i);
        v.push_back({i, cnt});
        maxCnt = max(maxCnt, cnt);
    }

    for (auto& p : v)
    {
        if (p.second == maxCnt)
            cout << p.first << " ";
    }

나머지 코드는 잘 짜셨습니다. ㅎㅎ

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.

-1

인프런 AI 인턴

안녕하세요, 인프런 AI 인턴입니다. 요가인님.

작성하신 코드를 보면, 주어진 문제는 그래프에서 특정 노드에서 시작했을 때 가장 많은 노드에 영향을 미칠 수 있는 노드를 찾는 문제로 보입니다. 현재 작성하신 코드의 로직은 각 노드에서 DFS(깊이 우선 탐색)를 실행하여 도달할 수 있는 노드의 수를 계산하고, 그 결과에 따라 많은 영향을 미치는 노드를 출력하는 것입니다.

하지만, 입력이 많은 경우나 특정 그래프 구조에서는 시간 초과가 발생할 수 있고, 예상치 못한 결과를 출력할 수 있습니다. 특히 다음 사항을 고려해 보시면 좋습니다:

DFS 재귀 호출 제한: 노드의 수가 많을수록 DFS의 깊이가 깊어질 수 있습니다. 이는 호출 스택의 크기 제한과 관련이 있기 때문에, C++의 기본 스택 크기로 인해 스택 오버플로우가 발생할 수 있습니다.
시간 복잡도 최적화: dfs를 각 노드마다 실행하고 있는데, 이로 인해 시간 복잡도가 O(N * (N + M))이 되어 비효율적입니다. 문제에 따라 일반적인 DFS 보다는 BFS로 최적화하는 것이 좋을 수도 있습니다. 또한, 여러 노드 간의 도달 가능성을 미리 계산해 두면 좋습니다.
그래프의 입력 검사: 주어진 입력 데이터가 사이클을 형성할 수 있는지, 데이터 범위는 어떻게 제한되는지를 확인해야 하며, 이로 인해 예기치 못한 동작이 발생할 수 있습니다.
결과 값의 출력 순서: 현재 작성하신 코드에서 결과를 정렬하여 출력하고 있는데, 필요한 경우 직접적인 오름차순 정렬이 필요할 수 있습니다. sort(v.begin(), v.end(), cmp)에서는 두 번째 값을 기준으로 내림차순하고 있습니다.

아래는 참고할 만한 존재하는 질문 링크입니다. 같은 문제를 해결하려는 다른 분의 경험을 확인하실 수 있습니다:
- DFS에서 visited의 존재는 속도를 더 빠르게 해주나요? (1325번)
- 2-S 문제 관련해서 유사한 질문

추가적으로, 도달 가능한 노드의 수를 계산하는 부분에서 메모이제이션이나 동적 계획법(DP)을 활용하여 중간 결과를 재사용하면 성능을 더욱 개선할 수 있습니다.

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다. 현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.

인프런 커뮤니티 질문&답변

어디 부분이 잘못되었는지 잘 모르겠습니다.