# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1055414 | Hazard | Closing Time (IOI23_closing) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int MAX_N = 100005;
const int MAX_M = 200005;
vector<int> graph[MAX_N];
int U[MAX_M], V[MAX_M], W[MAX_M];
int N, M, K, X, Y;
int dfs(int node, int time, int parent, vector<int>& closing_times) {
int reachable = 0;
for (int i = 0; i < graph[node].size(); i++) {
int next_node = graph[node][i];
if (next_node == parent) continue;
int next_time = time + W[graph[node][i]];
if (next_time <= closing_times[next_node]) {
reachable += dfs(next_node, next_time, node, closing_times);
}
}
return reachable + 1;
}
int compute_convenience_score(vector<int>& closing_times) {
int score = 0;
score += dfs(X, 0, -1, closing_times);
score += dfs(Y, 0, -1, closing_times);
return score;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cin >> N >> M >> K >> X >> Y;
for (int i = 0; i < M; i++) {
cin >> U[i] >> V[i] >> W[i];
graph[U[i]].push_back(i);
graph[V[i]].push_back(i);
}
int max_score = 0;
for (int mask = 0; mask < (1 << N); mask++) {
vector<int> closing_times(N);
for (int i = 0; i < N; i++) {
closing_times[i] = (mask >> i) & 1;
}
int sum_closing_times = 0;
for (int i = 0; i < N; i++) {
sum_closing_times += closing_times[i];
}
if (sum_closing_times <= K) {
int score = compute_convenience_score(closing_times);
max_score = max(max_score, score);
}
}
cout << max_score << endl;
return 0;
}