QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#420896 | #6523. Escape Plan | shepherd | AC ✓ | 2038ms | 71628kb | C++20 | 2.1kb | 2024-05-25 02:58:22 | 2024-05-25 02:58:23 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
static constexpr const ll inf = 1e15;
int main() {
int t;
cin >> t;
while (t--) {
int n,m,k;
cin >> n >> m >> k;
vector<bool> is_exit(n);
for (int i = 0; i<k; i++) {
int e;
cin >> e;
is_exit[--e] = true;
}
vector<int> d(n);
for (int i = 0; i<n; i++) {
cin >> d[i];
}
vector<vector<pair<int,ll>>> graph(n);
for (int i = 0; i<m; i++) {
int x,y;
ll w;
cin >> x >> y >> w;
graph[--x].emplace_back(--y, w);
graph[y].emplace_back(x, w);
}
vector<ll> ret(n, inf);
vector<bool> marked(n, false);
vector<multiset<ll>> dd(n);
auto cmp = [&](int x, int y) {
auto dx = *prev(end(dd[x]));
auto dy = *prev(end(dd[y]));
if (dx != dy) return dx < dy;
return x < y;
};
set<int, decltype(cmp)> q(cmp);
for (int i = 0; i<n; i++) {
if (is_exit[i]) {
dd[i].insert(0);
q.insert(i);
d[i] = 0;
}
}
while (!q.empty()) {
auto curr = *q.begin();
q.erase(q.begin());
marked[curr] = true;
ret[curr] = *prev(end(dd[curr]));
for (const auto& [neighbor, weight] : graph[curr]) {
if (marked[neighbor]) continue;
if (int(dd[neighbor].size()) > d[neighbor]) {
q.erase(neighbor);
}
dd[neighbor].insert(ret[curr] + weight);
if (int(dd[neighbor].size()) > d[neighbor] + 1) {
dd[neighbor].erase(prev(end(dd[neighbor])));
}
if (int(dd[neighbor].size()) > d[neighbor]) {
q.insert(neighbor);
}
}
}
if (ret[0] >= inf) cout << -1 << '\n';
else cout << ret[0] << '\n';
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3512kb
input:
2 3 4 1 3 1 1 1 1 2 1 1 2 2 2 3 1 2 3 2 3 2 2 2 3 2 0 0 1 2 1 1 3 1
output:
4 -1
result:
ok 2 number(s): "4 -1"
Test #2:
score: 0
Accepted
time: 2038ms
memory: 71628kb
input:
100 100 1808 2 94 47 3 3 0 2 4 3 3 4 0 0 2 2 2 3 2 4 0 2 3 4 4 2 0 3 4 3 1 0 2 1 2 2 0 3 4 4 4 1 2 2 3 1 0 0 3 1 4 2 1 3 3 4 3 0 4 1 0 3 2 1 4 4 1 3 2 3 3 3 3 1 0 3 0 4 3 1 0 4 0 4 4 1 2 0 0 4 1 3 3 3 0 2 2 1 1 2 3 4 1 2 72 29 1138 59 78 2398 95 5 1610 32 46 4176 36 99 8143 100 69 413 61 58 1595 9 9...
output:
5109 1021 3293 4646 3796 3394 1884 6772 2329 2067 3296 2809 865 4249 2241 3792 2135 2544 3343 1775 10602 4677 1700 2150 7071 14055 3368 2322 1113 1980 3067 1617 1702 -1 2879 6265 2065 2810 2289 3001 402 3769 18118 6874 7879 3823 -1 510 2636 10564 -1 3166 3615 7526 5549 1261 3302 270 4440 1998 3350 3...
result:
ok 100 numbers