1. 문제 설명
- 문제 링크
1번 컴퓨터와 연결된 컴퓨터의 수 찾기
조건
첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어진다. 이어서 그 수만큼 한 줄에 한 쌍씩 네트워크 상에서 직접 연결되어 있는 컴퓨터의 번호 쌍이 주어진다.
2. 풀이
- 재귀를 활용한 dfs
3. 코드
- c++
#define MAX(A,B) A > B ? A : B;
#define MIN(A,B) A < B ? A : B;
#include<iostream>
#include<string.h>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
vector<int> networks[101];
bool visit[101];
int answer;
void dfs(int V) {
answer++;
visit[V] = true;
for (int i = 0; i < networks[V].size(); i++) {
int node = networks[V][i];
if (!visit[node]) dfs(node);
}
}
int main(int argc, char** argv)
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int N, M;
cin >> N >> M;
for (int i = 0; i < M; i++) {
int a, b;
cin >> a >> b;
networks[a].push_back(b);
networks[b].push_back(a);
}
answer = 0;
memset(visit, false, sizeof(visit));
dfs(1);
cout << answer-1 << endl;
return 0;//정상종료시 반드시 0을 리턴해야합니다.
}
- python
N = int(input())
M = int(input())
networks = [[0] * (N+1) for _ in range(N+1)]
visit = [False] * (N+1)
for _ in range(M):
_from, _to = map(int, input().split())
networks[_from][_to] = 1
networks[_to][_from] = 1
answer = 0
def dfs(n, answer):
visit[n] = True
for i in range(1, N+1):
if networks[n][i] == 1 and visit[i] == False:
answer = dfs(i, answer + 1)
return answer
print(dfs(1, answer))
4. 결과
간단한 문제라 생략!