준형이와 친구들의 왕복시간들 중 최대가 최소가 되는 값을 찾아야 하는데, 이 문항을 대충 읽어서 tc를 이해를 못하고 헤맸다...
플로이드 돌리고 도시 마다 친구 목록 돌면서 왕복시간의 최대값 기준으로 최솟값을 찾으면 된다.
예전에는 별 생각 없었는데, 자바스크립트를 많이 하다 보니 ps할 때 나의 c++ 코드는 참 더럽다.
#include <bits/stdc++.h>
#define fastio ios::sync_with_stdio(0), cin.tie(0)
using namespace std;
using ll = long long;
using pii = pair<int, int>;
#define all(v) v.begin(), v.end()
int dx[8] = {1, 0, -1, 0, 1, -1, -1, 1};
int dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};
int arr[205][205];
int friends[205];
int n, m, K;
void input() {
cin >> n >> m;
for(int i = 0; i < m; ++i) {
int a, b, c; cin >> a >> b >> c;
arr[a][b] = c;
}
cin >> K;
for(int i = 0; i < K; ++i) cin >> friends[i];
}
int main() {
fastio;
memset(arr, 0x3f, sizeof(arr));
input();
for(int i = 1; i <= n; ++i) arr[i][i] = 0;
for(int k = 1; k <= n; ++k) {
for(int i = 1; i <= n; ++i) {
for(int j = 1; j <= n; ++j) {
arr[i][j] = min(arr[i][j], arr[i][k] + arr[k][j]);
}
}
}
vector<int> ans;
int mCost = 1e9 + 1;
for(int i = 1; i <= n; ++i) {
bool invalid = false;
int tmp = 0;
for(int j = 0; j < K; ++j) {
if(arr[friends[j]][i] > 1e9 || arr[i][friends[j]] > 1e9) {
invalid = true;
break;
}
tmp = max(tmp, arr[friends[j]][i] + arr[i][friends[j]]);
}
if(invalid) continue;
if(mCost > tmp) {
ans.clear();
ans.push_back(i);
mCost = tmp;
}
else if(mCost == tmp) {
ans.push_back(i);
}
}
for(auto num : ans) cout << num << " ";
return 0;
}
백준 21938 영상처리 혼내주기 (0) | 2022.03.22 |
---|---|
백준 21937 작업 혼내주기 (0) | 2022.03.22 |
백준 21939 문제 추천 시스템 Version 1 혼내주기 (0) | 2022.03.15 |
백준 21924 도시 건설 혼내주기 (0) | 2022.03.03 |
백준 21923 곡예 비행 혼내주기 (0) | 2022.03.03 |
댓글 영역