티스토리 뷰

https://school.programmers.co.kr/learn/courses/30/lessons/42861
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
이번 문제는 각 섬들(이하 노드, 노드들)과 이들을 이을 때 드는 비용들이 주어질 때, 모든 노드들을 이을 수 있는 가장 적은 비용을 구하는 문제이다. 이 문제는 딱 보자마자 크루스칼 알고리즘 문제라는 생각이 들었다. 크루스칼 알고리즘이란 가장 적은 비용으로 모든 노드들을 연결할 때 이용할 수 있는 알고리즘이기 때문이다. 즉, 이 문제는 크루스칼 알고리즘의 노드를 섬으로 바꿨을 뿐인 문제이다. 그런데 크루스칼 알고리즘을 이용하기 위해서는 비용들이 오름차순으로 정렬되어 있어야 한다. 하지만 주어진 costs의 값은 그렇지 않다. 그렇기 때문에 미리 정렬을 해주어야 한다.
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
#define MAX 101
int parentNode[MAX];
bool Compare(vector<int> a, vector<int> b)
{
return a[2] < b[2];
}
int ParentNode(int node)
{
if (parentNode[node] == node)
return parentNode[node];
return parentNode[node] = ParentNode(parentNode[node]);
}
void MakeSame(int a, int b)
{
a = ParentNode(a);
b = ParentNode(b);
if (a < b)
parentNode[b] = a;
else
parentNode[a] = b;
}
bool IsSame(int a, int b)
{
a = ParentNode(a);
b = ParentNode(b);
return a == b;
}
int solution(int n, vector<vector<int>> costs) {
int answer = 0;
for (int node = 0; node < n; ++node)
parentNode[node] = node;
sort(costs.begin(), costs.end(), Compare);
for (int i = 0; i < costs.size(); ++i)
{
int nodeA = costs[i][0];
int nodeB = costs[i][1];
int cost = costs[i][2];
if (IsSame(nodeA, nodeB) == false)
{
MakeSame(nodeA, nodeB);
answer += cost;
}
}
return answer;
}
참고
- 크루스칼 알고리즘
'알고리즘 문제' 카테고리의 다른 글
프로그래머스 - 불량 사용자 (C++) (0) | 2023.05.31 |
---|---|
프로그래머스 - 가장 먼 노드 (C++) (0) | 2023.05.09 |
프로그래머스 - 소수 찾기 (C++) (0) | 2023.04.22 |
백준 - 1543번 문서 검색 (C++) (0) | 2023.04.10 |
백준 - 17608번 막대기 (C++) (0) | 2023.04.08 |
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 브루트포스
- 알고리즘
- 타입스크립트
- C++
- Redux
- Next.js
- 백준
- SQL
- NextJS
- typescript
- 코드스테이츠
- 자바스크립트
- 리액트
- 햄버거버튼
- 다이나믹프로그래밍
- 비트마스킹
- 구현
- 넥스트js
- 동적계획법
- BFS
- aws
- 카카오맵
- react router
- 스택
- 순열
- CSS
- react
- themoviedb
- 프로그래머스
- 완전탐색
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | ||
6 | 7 | 8 | 9 | 10 | 11 | 12 |
13 | 14 | 15 | 16 | 17 | 18 | 19 |
20 | 21 | 22 | 23 | 24 | 25 | 26 |
27 | 28 | 29 | 30 | 31 |
글 보관함