QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#420884 | #8038. Hammer to Fall | VeryAmazed | ML | 0ms | 3652kb | C++20 | 1.6kb | 2024-05-25 02:52:07 | 2024-05-25 02:52:07 |
Judging History
answer
// Right side, 7th row, inner, Johns Hopkins University
#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for (int i = a; i < (b); +i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
int main() {
cin.tie(0)->sync_with_stdio(0);
cin.exceptions(cin.failbit);
int n, m, q; cin >> n >> m >> q;
vector<int> residents(n); for (int &x : residents) cin >> x;
vector adj(n, vector<pair<int,int>>());
while (m--) {
int u, v, w; cin >> u >> v >> w; u--, v--;
adj[u].emplace_back(v, w);
adj[v].emplace_back(u, w);
}
vector<pair<int,int>> parent(n);
vector children(n, vector<pair<int,int>>());
vector<multiset<long long>> cost(n);
auto build = [&](auto self, int u, int p, int w) -> void {
parent[u] = {p, w};
for (auto [v,w1] : adj[u]) if (v != p) self(self, v, u, w1), cost[u].emplace(w1);
};
build(build, 0, 0, 0);
vector<long long> cur_cost(n);
vector<int> hammers(q); for (int &x : hammers) cin >> x, x--; reverse(hammers.begin(),hammers.end());
for (int h : hammers) {
if (h) {
auto it = cost[parent[h].first].find(cur_cost[h] + parent[h].second);
cost[parent[h].first].erase(it);
}
long long mn = LLONG_MAX;
if (h) mn = cur_cost[parent[h].first] + parent[h].second;
if (cost[h].size()) mn = min(mn, *cost[h].begin());
cur_cost[h] = mn;
if (h) {
cost[parent[h].first].emplace(cur_cost[h] + parent[h].second);
}
}
const int M = 998244353;
int ans = 0;
for (int i = 0; i < n; i++) {
ans = (ans + 1ll * residents[i] * (cur_cost[i] % M)) % M;
}
cout << ans << '\n';
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3652kb
input:
3 2 2 1 1 1 2 3 10 1 2 1 3 2
output:
12
result:
ok single line: '12'
Test #2:
score: 0
Accepted
time: 0ms
memory: 3652kb
input:
2 1 10 5000 5000 1 2 10000 1 2 2 1 2 2 1 1 1 2
output:
550000000
result:
ok single line: '550000000'
Test #3:
score: -100
Memory Limit Exceeded
input:
10 10 10 5 14 99 14 18 4 58 39 48 60 2 4 4 6 9 56 10 8 34 7 5 96 1 3 26 3 7 92 6 8 4 5 1 72 7 6 39 7 2 93 8 8 9 10 2 2 5 9 2 3