프로그래머스 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;
	}

view raw 네트워크.cpp hosted with ❤ by GitHub

 

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

view raw 네트워크.py hosted with ❤ by GitHub

• for i in range(len(computers[cur]))

배열 길이로 할 때 len으로 감싸줘서 길이 구하고 정수는 range로 감싸주기

 

• for i in range(n):

얘도 마찬가지. for i in n 이렇게 하면 typeerror 발생한다.