끄적끄적
프로그래머스 lv3 C++, python [네트워크] - dfs 본문
출처 : https://programmers.co.kr/learn/courses/30/lessons/43162
문제 설명
네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.
컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.
제한사항
- 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.
- 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.
- i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computers[i][j]를 1로 표현합니다.
- computer[i][i]는 항상 1입니다.
입출력 예
n computers return
3 | [[1, 1, 0], [1, 1, 0], [0, 0, 1]] | 2 |
3 | [[1, 1, 0], [1, 1, 1], [0, 1, 1]] | 1 |
입출력 예 설명
예제 #1
아래와 같이 2개의 네트워크가 있습니다.

예제 #2
아래와 같이 1개의 네트워크가 있습니다.

1. c++
#include <string> | |
#include <vector> | |
using namespace std; | |
bool dfs(vector<vector<int>> &computers, int cur){ | |
//이미 순회한 노드라면 리턴(재귀함수 탈출조건) | |
if(!computers[cur][cur]) | |
return false; | |
computers[cur][cur]=0; | |
for(int i=0; i<computers[cur].size(); i++){ | |
if(computers[cur][i]) | |
dfs(computers, i); | |
} | |
return true; | |
} | |
int solution(int n, vector<vector<int>> computers) { | |
int answer = 0; | |
for(int i=0; i<computers.size(); i++){ | |
if(computers[i][i] && dfs(computers, i)) answer++; | |
} | |
return answer; | |
} |
dfs 문제
연결 여부와 네트워크 형성을 체크하기 위해, 이미 방문한 노드는 0으로 값을 바꾸어 제외시켜준다.
연결된 노드가 있다면 다시 재귀를 들어가고, 더 이상 연결된 노드가 없다면 true를 반환하게 되면서 네트워크 개수를 1개 추가시켜주게 된다.
이미 방문한 노드라면 0이기 때문에 false를 리턴하여 개수가 증가하지 않게 한다.
인자를 넘겨줄 때 &를 사용해야 답이 맞고, &를 사용하지 않으면 노드 개수 전체(예제에서는 3)가 출력되던데
잘은 모르겠지만 &를 사용하지 않으면 값을 바꿔줄 때 제대로 반영이 안돼서 그런 것 같다..
2. python
def dfs(computers, cur): | |
if computers[cur][cur]==0: | |
return False | |
computers[cur][cur]=0 | |
for i in range(len(computers[cur])): | |
if computers[cur][i]: | |
dfs(computers,i) | |
return True | |
def solution(n, computers): | |
answer = 0 | |
for i in range(n): | |
if computers[i][i] and dfs(computers,i): | |
answer+=1 | |
return answer |
- for i in range(len(computers[cur]))
배열 길이로 할 때 len으로 감싸줘서 길이 구하고 정수는 range로 감싸주기
- for i in range(n):
얘도 마찬가지. for i in n 이렇게 하면 typeerror 발생한다.

'코테준비' 카테고리의 다른 글
프로그래머스 level2 [소수 찾기] - 완탐 (0) | 2022.03.28 |
---|---|
프로그래머스 lv2 [타겟 넘버] C++, python - bfs/dfs (0) | 2022.03.26 |
프로그래머스 level1 [모의고사] c++, python - 완전탐색 (0) | 2022.03.26 |
프로그래머스 lv2 [전화번호 목록] C++, python - 해시 (0) | 2022.03.22 |
프로그래머스 lv1 [완주하지 못한 선수] C++, python - 해시 (0) | 2022.03.21 |