백준 21940번 - C++

#include <bits/stdc++.h>
using namespace std;

const int INF = 0x3f3f3f3f;
int v[205][205];
vector<int> vv;
int ans[205];
int n, e,kk;
int a, b, c;
int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> n >> e;
	for (int i = 1; i <= n; i++)
		fill(v[i], v[i] + 1 + n, INF);
	for (int i = 0; i < e; i++) {
		cin >> a >> b >> c;
		v[a][b] = min(v[a][b], c);
	}
	for (int i = 1; i <= n; i++) v[i][i] = 0;
	cin >> kk;
	for (int i = 0; i < kk; i++) {
		int aa;
		cin >> aa;
		vv.push_back(aa);
	}

	for (int k = 1; k <= n; k++)
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++)
				v[i][j] = min(v[i][j], v[i][k] + v[k][j]); // if 문으로 바꾸면 시간을 줄일 수 있다.


	int flag2 = INF;
	for (int i = 1; i <= n;i++) {
		int flag = 0;
		for (auto j:vv) {
			flag =max(flag, v[j][i]+v[i][j]);
		}
		ans[i] = flag;
		flag2 = min(flag, flag2);
	}

	for (int i = 1; i <= n; i++) {
		if (ans[i] == flag2) cout << i<<' ';
	}
	

}