# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
965397 | MisterReaper | Harbingers (CEOI09_harbingers) | C++17 | 1072 ms | 19728 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using i64 = long long;
constexpr int maxN = 1E5 + 5;
constexpr i64 INF = 1E18;
constexpr __int128 ONE = 1;
int N;
std::vector<std::pair<int, int>> adj[maxN];
int S[maxN], V[maxN], par[maxN], D[maxN];
i64 dp[maxN];
struct line {
i64 a, b;
line(i64 _a, i64 _b) : a(_a), b(_b) {}
i64 eval(i64 x) {
return 1LL * a * x + b;
}
long double intersect(line rhs) {
return (b - rhs.b) / (rhs.a - a);
}
};
struct CHTrick {
std::deque<int> before;
std::deque<line> dq;
std::deque<line> rem;
CHTrick() {}
int size() { return (int) dq.size(); }
i64 query(i64 x) {
int l = 0, r = size() - 1;
while(r - l > 3) {
int m = (l + r) >> 1;
if(dq[m].eval(x) >= dq[m + 1].eval(x)) {
l = m;
} else {
r = m;
}
}
i64 res = INF;
for(int i = l; i <= r; i++) {
res = std::min(res, dq[i].eval(x));
}
return res;
}
void add(line l) {
before.emplace_back(size());
while(size() >= 2 && ONE * (l.b - dq[0].b) * (dq[1].a - dq[0].a) <= ONE * (dq[0].a - l.a) * (dq[0].b - dq[1].b)) {
rem.push_back(dq[0]);
dq.pop_front();
}
dq.push_front(l);
}
void rollback() {
dq.pop_front();
std::deque<line> save;
while((int) save.size() + size() != before.back()) {
save.push_front(rem.back());
rem.pop_back();
}
while(!save.empty()) {
dq.push_front(save.back());
save.pop_back();
}
before.pop_back();
}
line operator[](int i) {
return dq[i];
}
} CHT;
void dfs(int v) {
if(v != 1) {
dp[v] = CHT.query(V[v]) + 1LL * D[v] * V[v] + S[v];
}
CHT.add({-D[v], dp[v]});
for(auto [u, w] : adj[v]) {
if(u == par[v]) {
continue;
}
D[u] = D[v] + w;
par[u] = v;
dfs(u);
}
CHT.rollback();
}
signed main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
std::cin >> N;
for(int i = 1; i <= N - 1; i++) {
int A, B, W;
std::cin >> A >> B >> W;
adj[A].emplace_back(B, W);
adj[B].emplace_back(A, W);
}
for(int i = 2; i <= N; i++) {
std::cin >> S[i] >> V[i];
}
std::fill_n(&dp[1], N, INF);
dp[1] = 0;
par[1] = 1;
dfs(1);
for(int i = 2; i <= N; i++) {
std::cout << dp[i] << " \n"[i == N];
}
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |