16562번 친구비

2018년 서강대학교 프로그래밍 콘테스트(Mater) B번 문제

대회 : https://www.acmicpc.net/category/detail/1961

문제 : https://www.acmicpc.net/problem/16562

DFS를 통해 같은 관계인 친구들 중에서 최소의 돈을 구한다.

그리고 배열 r에 관계에 해당하는 최소의 돈을 저장한다.(r=관계)

#include <iostream>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <cstring>
#include <string>
#include <math.h>
#include <algorithm>
#include <map>
 
using namespace std;
 
vector<int> v[10001];
 
int N = 0;
 
int Money[10001] = { 0, };
 
int r[10001] = { 0, };
 
int MIN = 0;
 
bool visit[10001] = { false, };
 
void DFS(int start)
{
    visit[start] = true;
 
    MIN = min(MIN, Money[start]);
 
    for (auto next : v[start])
    {
        if (!visit[next])
        {
            DFS(next);
        }
    }
}
 
int main(void)
{
    int money = 0, M = 0, k = 0;
 
    scanf("%d %d %d", &N, &M, &k);
 
    for (int i = 1; i <= N; i++)
    {
        scanf("%d", &Money[i]);
    }
 
    for (int i = 0; i < M; i++)
    {
        int s = 0, e = 0;
 
        scanf("%d %d", &s, &e);
 
        v[s].push_back(e);
        v[e].push_back(s);
    }
 
    int index = 0;
 
    for (int i = 1; i <= N; i++)
    {
        if (!visit[i])
        {
            MIN = INT32_MAX;
 
            DFS(i);
 
            r[index++] = MIN;
        }
    }
 
    bool isStop = false;
 
    int sum = 0;
 
    for (int i = 0; i < index; i++)
    {
        k -= r[i];
        sum += r[i];
 
        if (k < 0)
        {
            isStop = true;
            break;
        }
    }
 
    if (isStop) printf("Oh no\n");
    else printf("%d\n", sum);
 
    return 0;
}