[백준/c++] BOJ 21278 - 호석이 두 마리 치킨

문제 설명

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

21278번: 호석이 두 마리 치킨

 

문제 설명

도시는 N개의 건물과 M개의 도로로 이루어져 있다. 건물은 1번부터 N번의 번호를 가지고 있다. i번째 도로는 서로 다른 두 건물 Ai와 Bi 사이를 1시간에 양방향으로 이동할 수 있는 도로이다. N개의 건물 중, 2개의 건물을 골라 치킨집을 지으려고 한다. 단, "모든 건물에서 가장 가까운 치킨집까지 왕복하는 최단 시간의 총합"을 최소화할 수 있는 건물을 고른다.

이러한 건물 조합이 여러 개라면, 건물 번호 중 작은 게 더 작을수록, 작은 번호가 같다면 큰 번호가 더 작을수록 좋은 건물 조합이다.

N (2 <= N <= 100)

M (N - 1 <= M <= N * (N - 1) / 2)

Ai, Bi (1 <= Ai, Bi <= N (Ai != Bi))

해결

이중포문으로 n, n을 돌며 모든 좌표에서 bfs를 해 주었다. 

 

bfs로 치킨집에서 다른 건물을 방문할 때 걸리는 최단 시간을 구했다.

왕복 시간은 걸리는 시간에 2배를 해 주면 되므로 그저 모든 건물을 방문하는데 최단 시간이 걸리는 건물 조합을 출력하면 된다.

코드

#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;

int n, m;
int a, b;
int ansa, ansb, ans = 987654321;

vector<int> v[101];

void bfs(int i, int j) {
	int visited[101] = { 0 }, tmp = 0;
	queue<pair<int, int>> q; //좌표, 시간
	q.push({ i, 0 });
	q.push({ j ,0 });
	visited[i] = 0;
	visited[j] = 0;

	while (!q.empty()) {
		auto cur = q.front(); q.pop();
		for (int a : v[cur.x]) {
			if (visited[a] || a == i || a == j) continue;
			visited[a] = cur.y + 1;
			tmp += visited[a];
			q.push({ a, cur.y + 1 });
		}
	}
	
	if (tmp < ans) {
		ansa = i, ansb = j, ans = tmp;
	}

}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		cin >> a >> b;
		v[a - 1].push_back(b - 1);
		v[b - 1].push_back(a - 1);
	}

	for (int i = 0; i < n; i++) {
		for (int j = i + 1; j < n; j++)
			bfs(i, j);
	}

	cout << ansa + 1 << " " << ansb + 1 << " " << ans * 2;
}