inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

2-S

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

해결된 질문

49

요가인

작성한 질문수 35

0

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

#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

0

큰돌

안녕하세요 가인님 ㅎㅎ

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

    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(깊이 우선 탐색)를 실행하여 도달할 수 있는 노드의 수를 계산하고, 그 결과에 따라 많은 영향을 미치는 노드를 출력하는 것입니다.

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

  1. DFS 재귀 호출 제한: 노드의 수가 많을수록 DFS의 깊이가 깊어질 수 있습니다. 이는 호출 스택의 크기 제한과 관련이 있기 때문에, C++의 기본 스택 크기로 인해 스택 오버플로우가 발생할 수 있습니다.

  2. 시간 복잡도 최적화: dfs를 각 노드마다 실행하고 있는데, 이로 인해 시간 복잡도가 O(N * (N + M))이 되어 비효율적입니다. 문제에 따라 일반적인 DFS 보다는 BFS로 최적화하는 것이 좋을 수도 있습니다. 또한, 여러 노드 간의 도달 가능성을 미리 계산해 두면 좋습니다.

  3. 그래프의 입력 검사: 주어진 입력 데이터가 사이클을 형성할 수 있는지, 데이터 범위는 어떻게 제한되는지를 확인해야 하며, 이로 인해 예기치 못한 동작이 발생할 수 있습니다.

  4. 결과 값의 출력 순서: 현재 작성하신 코드에서 결과를 정렬하여 출력하고 있는데, 필요한 경우 직접적인 오름차순 정렬이 필요할 수 있습니다. sort(v.begin(), v.end(), cmp)에서는 두 번째 값을 기준으로 내림차순하고 있습니다.

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

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

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

2-I) 왜 이 문제가 그래프이론 카테고리에 있는지 잘 모르겠습니다.

0

17

2

2주차 개념#12 트리 순회

0

19

2

백준사이트가 종료된다고 합니다.

0

230

2

백준 서비스 종료

9

731

1

sk 하이닉스 코테 대비

0

354

2

3-G 최댓값 질문

0

50

1

모듈러 연산 값이 10이 아닌 경우도 있지 않나요?

0

82

2

3-I 코드 질문드립니다.

0

61

2

3-N 질문 있습니다.

0

66

2

학습방법

0

100

2

4-H 질문 있습니다 (코드 리뷰)

0

66

2

코딩테스트 어디까지 준비해야 하는지 질문이 있습니다.

0

164

2

2-O 반례가 무엇일지 어떤 부분이 틀렸는지 잘 모르겠습니다.

0

69

2

2주차 개념 #4-2. 인접행렬 질문있습니다.

0

63

2

1-A 문제풀이 후 궁금한 점이 생겨서 질문드립니다.

0

49

2

조합 재귀 풀이 확인 해주시면 감사하겠습니다.

0

67

2

함수별 시간복잡도

0

72

2

3-h 질문입니다.

0

49

1

안녕하세요 선생님. 시간 복잡도 4번 질문있습니다.

0

52

2

1-I 문제 질문 드립니다.

0

76

2

2-P 질문입니다.

0

56

1

mac에서 시작하기 관련

0

88

2

5-Q 질문

0

63

2

풀이 코드 질문

0

63

2