글 읽기 - 2606(바이러스) 질문

whgkdrjs4321 4년 전 0

안녕하세요 BFS/DFS를 공부하고 있는 사람입니다.

2606(바이러스)번을 DFS를 이용해서 풀어보았는데 채점을 하면 틀렸다고 나옵니다. 그런데 어디서 틀린 것인지 잘 모르겠습니다.

혹시 어느 부분에서 틀렸는지 예시와 제가 작성한 코드를 어떻게 혹은 어느 부분을 고쳐야 하는지 고수님들의 도움을 받고싶습니다.

#include <iostream>
#include <vector>
using namespace std;
//DFS로 풀어보기

vector<int> v[101];					//서로 연결되어 있는지 확인할 배열
int visited[101] = { 0, };			//방문한 것을 표현하는 배열(0 : 방문X, 1 : 방문)
int cnt = 0;						//웜 바이러스에 걸린 컴퓨터의 수

void dfs(int a)
{
	if (visited[a])					//이미 방문한 경우 -> 탈출!
		return;

	visited[a] = 1;					//이제 방문한 것으로 만든다!
	cnt++;							//방문하면 1씩 증가!
	
	for (int i = 0; i < v[a].size(); i++)	//v[a]에 있는 모든 값들을 방문!
	{
		int b = v[a][i];

		dfs(b);
	}
}

int main()
{
	int n, m;							//n : 컴퓨터의 수, m : 네트워크상에 직접 연결된 컴퓨터의 번호 쌍
	int a, b;							//컴퓨터 쌍을 입력할 변수
	

	cin >> n >> m;

	for (int i = 0; i < m; i++)
	{
		cin >> a >> b;

		v[a].push_back(b);
		//v[b].push_back(a);
	}

	dfs(1);

	cout << cnt - 1 << endl;

	return 0;
}