[ 백준 ] 13418번 - 학교 탐방하기 (C++)

https://www.acmicpc.net/problem/13418

13418번: 학교 탐방하기

 

 

[ 문제풀이 ]

 

1. node struct를 만들고, node arr [ 500000 ] 배열을 만들어 각각의 A, B, C를 저장합니다. 이때, 오르막길이 0이므로 0이 입력되면 C = 1, 1이 입력되면 C = 0으로 저장합니다.

 

2. arr를 오름차순으로 정렬하고, vect 배열을 초기화한 후 Union-Find를 이용하여 각 노드들을 이어주고, cost를 max에 더해줍니다.

 

3. arr를 내림차순으로 정렬하고, vect 배열을 초기화한 후 Union-Find를 이용하여 각 노드들을 이어주고, cost를 min에 더해줍니다.

 

4. max * max - min * min을 출력해줍니다.

 

[ 소스코드 ]

#include<iostream>
#include<queue>
#include<algorithm>
#define ll long long
 
using namespace std;
 
struct node {
    int A;
    int B;
    int cost;
};
 
struct Greater {
    bool operator()(node right, node left) {
        return left.cost < right.cost;
    }
};
 
struct Less {
    bool operator()(node right, node left) {
        return left.cost > right.cost;
    }
};
 
int N, M;
int vect[1001];
node arr[500000];
 
int Find(int num)
{
    if (vect[num] == num) return num;
    return vect[num] = Find(vect[num]);
}
 
void Union(int a, int b)
{
    int pa = Find(a);
    int pb = Find(b);
 
    vect[pb] = pa;
}
 
ll fatigue(ll a)
{
    ll ret = a;
 
    for (int i = 1; i <= M; i++) {
        const int A = arr[i].A;
        const int B = arr[i].B;
        const int cost = arr[i].cost;
 
        if (Find(A) != Find(B)) {
            Union(A, B);
            ret += cost;
        }
    }
 
    return ret;
}
 
int main()
{
    scanf("%d %d", &N, &M);
 
    for (int i = 0; i <= M; i++) {
        scanf("%d %d", &arr[i].A, &arr[i].B);
        int cost;
        scanf("%d", &cost);
        if (cost == 0) {
            arr[i].cost = 1;
        }
        else {
            arr[i].cost = 0;
        }
    }
 
    ll max, min;
 
    for (int i = 0; i <= N; i++) {
        vect[i] = i;
    }
 
    vect[1] = 0;
 
    sort(arr + 1, arr + M + 1, Greater());
 
    max = fatigue(arr[0].cost);
 
    for (int i = 0; i <= N; i++) {
        vect[i] = i;
    }
 
    vect[1] = 0;
 
    sort(arr + 1, arr + M + 1, Less());
 
    min = fatigue(arr[0].cost);
 
    printf("%lld", max * max - min * min);
}