프로그래머스: 전력망을 둘로 나누기⭐⭐

업데이트: 2024.07.10 Updated: 2024.07.10

카테고리: Programmers

태그: Algorithm, Blog, C++, Programmers

난이도 ⭐⭐

🧐 문제

문제 설명

n개의 송전탑이 전선을 통해 하나의 트리 형태로 연결되어 있습니다. 당신은 이 전선들 중 하나를 끊어서 현재의 전력망 네트워크를 2개로 분할하려고 합니다. 이때, 두 전력망이 갖게 되는 송전탑의 개수를 최대한 비슷하게 맞추고자 합니다.

송전탑의 개수 n, 그리고 전선 정보 wires가 매개변수로 주어집니다. 전선들 중 하나를 끊어서 송전탑 개수가 가능한 비슷하도록 두 전력망으로 나누었을 때, 두 전력망이 가지고 있는 송전탑 개수의 차이(절대값)를 return 하도록 solution 함수를 완성해주세요.

제한사항

n은 2 이상 100 이하인 자연수입니다.
wires는 길이가 n-1인 정수형 2차원 배열입니다.
- wires의 각 원소는 [v1, v2] 2개의 자연수로 이루어져 있으며, 이는 전력망의 v1번 송전탑과 v2번 송전탑이 전선으로 연결되어 있다는 것을 의미합니다.
- 1 ≤ v1 < v2 ≤ n 입니다.
- 전력망 네트워크가 하나의 트리 형태가 아닌 경우는 입력으로 주어지지 않습니다.

입출력 예

n	wires	result
9	`[[1,3],[2,3],[3,4],[4,5],[4,6],[4,7],[7,8],[7,9]]`	3
4	`[[1,2],[2,3],[3,4]]`	0
7	`[[1,2],[2,7],[3,7],[3,4],[4,5],[6,7]]`	1

입출력 예 설명

펼치기 / 접기

입출력 예 #1

다음 그림은 주어진 입력을 해결하는 방법 중 하나를 나타낸 것입니다.
4번과 7번을 연결하는 전선을 끊으면 두 전력망은 각 6개와 3개의 송전탑을 가지며, 이보다 더 비슷한 개수로 전력망을 나눌 수 없습니다.
또 다른 방법으로는 3번과 4번을 연결하는 전선을 끊어도 최선의 정답을 도출할 수 있습니다.

입출력 예 #2

다음 그림은 주어진 입력을 해결하는 방법을 나타낸 것입니다.
2번과 3번을 연결하는 전선을 끊으면 두 전력망이 모두 2개의 송전탑을 가지게 되며, 이 방법이 최선입니다.

입출력 예 #3

다음 그림은 주어진 입력을 해결하는 방법을 나타낸 것입니다.
3번과 7번을 연결하는 전선을 끊으면 두 전력망이 각각 4개와 3개의 송전탑을 가지게 되며, 이 방법이 최선입니다.

🔍힌트

순회를 통해 각 노드에서 자신과 자식 노드의 개수를 구한 후, 이 정보를 사용하여 문제를 쉽게 해결할 수 있다.

✏️ 나의 풀이

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <climits>

using namespace std;

// 간선 정보
int edges[101][101] = {0, };
// 각 노드에서의 자신과 자식 노드의 개수
int node_cnt[101] = {0, };

// 순회 함수
int traversal(int n, int cur) {
    // 자신 1개
    int cnt = 1;
    
    // 현재 노드와 연결된 모든 노드 찾기
    for(int i=1; i<=n; i++) {
        if(edges[cur][i] == 0) continue;
        
        // 재방문을 방지하기 위해, 연결 해제
        edges[cur][i] = 0;
        edges[i][cur] = 0;
        
        // 재귀를 통한 모든 자식 노드의 개수 구하고, 더하기
        cnt += traversal(n, i);
    }
    
    node_cnt[cur] = cnt;
    return cnt;
}

int solution(int n, vector<vector<int>> wires) {
    // 양방향 간선 정보 저장
    for(int i=0; i<wires.size(); i++) {
        int a = wires[i][0];
        int b = wires[i][1];
        edges[a][b] = 1;
        edges[b][a] = 1;
    }
    
    // 1번 노드부터 시작하여 완전 탐색 수행 
    traversal(n, 1);
    
    // 연결된 간선을 끊었을때, 두 전력망의 차이가 최소가 되는 값 찾기
    int min = INT_MAX;
    for(int i=1; i<=n; i++) {
        int cur = abs(n - 2*node_cnt[i]);
        if(min > cur) min = cur;
    }
    return min;
}

✔️ 다른 사람의 풀이

#include <bits/stdc++.h>

using namespace std;

int solution(int n, vector<vector<int>> wires) {
    vector<vector<int>> graph(n + 1);
    for(int i = 0; i < (int)wires.size(); i++) {
        int u = wires[i][0];
        int v = wires[i][1];
        graph[u].push_back(v);
        graph[v].push_back(u);
    }
    vector<int> siz(n + 1);
    function<void(int, int)> dfs = [&](int cur, int prev)  -> void {
        siz[cur] = 1;
        for(int nxt : graph[cur]) {
            if(nxt == prev) continue;
            dfs(nxt, cur);
            siz[cur] += siz[nxt];
        }
    };
    dfs(1, -1);
    int answer = INT_MAX;
    for(int i = 1; i <= n; i++) {
        for(int j : graph[i]) {
            int l = siz[j];
            int r = n - siz[j];
            answer = min(answer, abs(l - r));
        }
    }
    return answer;
}

🧐 분석

배열보다 벡터를 적극적으로 사용
함수가 아닌 람다 함수를 통해 코드를 축약
재방문하는 상황을 막아주기 위해 직접적으로 연결을 끊어준 것에 반해, prev 변수를 통해 간결하게 처리하고 있음.
범위 기반 for문을 사용하여 벡터에 쉽게 접근하고 있음.
min 함수를 사용하여 처리하고 있음.
<bits/stdc++.h>를 통해 자주 사용하는 표준 라이브러리 헤더를 한번에 Compile하고 있음. 하지만 장단점이 있는 방법임.

햄랑

프로그래머스: 전력망을 둘로 나누기⭐⭐

🧐 문제

문제 설명

제한사항