반응형
문제
N명의 사람이 있고, 여기서 세 사람 A, B, C를 고르려고 한다. 세 사람은 모두 친구여야 한다.
세 사람을 고르는 방법은 매우 많이 있을 수 있다. 이때, A의 친구 수 + B의 친구 수 + C의 친구 수가 최소가 되어야 한다. 친구 수의 합을 계산할 때, 세 사람은 빼고 계산해야 한다. 즉, A의 친구 수를 계산할 때, B와 C는 빼고 계산해야 하고, B의 친구 수를 계산할 때는 A와 C, C의 친구 수를 계산할 때는 A와 B를 빼고 계산해야 한다.
입력
첫째 줄에 사람의 수 N(3 ≤ N ≤ 4,000), 친구 관계의 수 M(0 ≤ M ≤ 4,000)이 주어진다. 둘째 줄부터 M개의 줄에는 친구 관계를 의미하는 두 정수 A, B가 주어진다. 친구 관계는 A와 B, 그리고 B와 A가 친구라는 것을 의미한다.
사람은 1번부터 N번까지 번호가 매겨져 있다. 같은 친구 관계가 두 번 이상 주어지는 경우는 없다.
출력
첫째 줄에 A의 친구 수 + B의 친구 수 + C의 친구 수의 최솟값을 출력한다. 만약, 문제 조건대로 세 사람을 고를 수 없는 경우에는 -1을 출력한다.
DFS를 3중 포문의 브루트포스로 변경시키면 됨.
#include<iostream>
#include<string>
#include<vector>
#include<queue>
#include<map>
#include<algorithm>
using namespace std;
int N, M;
bool rel[4001][4001];
int siz[4001];
int ans = 987654321;
bool flag;
int main(void)
{
cin >> N >> M;
for (int i = 0; i < M; i++)
{
int a, b;
cin >> a >> b;
rel[a][b] = true;
rel[b][a] = true;
siz[a]++;
siz[b]++;
}
for (int i = 1; i <= N; i++)
{
for (int j = i + 1; j <= N; j++)
{
if (rel[i][j])
{
for (int k = j + 1; k <= N; k++)
{
if (rel[i][k] && rel[j][k])
{
flag = true;
int res = siz[i] + siz[j] + siz[k] - 6;
if (ans > res)
{
ans = res;
}
}
}
}
}
}
if (!flag)
{
cout << -1 << endl;
}
else
{
cout << ans << endl;
}
}
반응형
'Algorithm' 카테고리의 다른 글
백준 1806번 부분합 (투포인터) (0) | 2020.07.29 |
---|---|
백준 2003번 수들의 합 2 (0) | 2020.07.29 |
백준 2210번 숫자판 점프 (0) | 2020.07.28 |
백준 17088번 등차수열 변환 (0) | 2020.07.27 |
백준 16938번 캠프 준비 (0) | 2020.07.23 |