Submission #985801

# Submission time Handle Problem Language Result Execution time Memory
985801 2024-05-18T21:24:47 Z OAleksa Harbingers (CEOI09_harbingers) C++14
70 / 100
1000 ms 21844 KB
#pragma GCC optimize("O3,unroll-loops")
#include <bits/stdc++.h>
#define f first
#define s second
using namespace std;
const int N = 1e5 + 69;
struct line {
	long long k, n;
	line(long long _k, long long _n) {
		k = _k;
		n = _n;
	}
	long long val(long long x) {
		return x * k + n;
	}
};
int n, s[N], w[N];
long long dp[N], dep[N];
vector<pair<int, int>> g[N];
vector<pair<line, int>> dq;
int dead[N], lst;
void dfs(int v, int p) {
	int l = 1, r = dq.size() - 1, id = 0;
	while (l <= r) {
		int mid = (l + r) / 2;
		if (dq[mid].f.val(w[v]) <= dq[mid - 1].f.val(w[v])) {
			id = mid;
			l = mid + 1;
		}
		else
			r = mid - 1;
	}
	dp[v] = dq[id].f.val(w[v]) + s[v] + 1ll * w[v] * dep[v];
	line ln = {-dep[v], dp[v]};
	int pr = lst, top = lst;
	while (dq.size() > 1 && (ln.k - dq.back().f.k) * (dq.back().f.n - dq[dq.size() - 2].f.n) <= (dq.back().f.n - ln.n) * (dq[dq.size() - 2].f.k - dq.back().f.k)) {
		dead[top] = dq.back().s;
		top++;
		lst++;
		dq.pop_back();
	}
	dq.push_back({ln, v});
	for (auto u : g[v]) {
		if (u.f == p)
			continue;
		dep[u.f] = dep[v] + u.s;
		dfs(u.f, v);
	}
	dq.pop_back();
	while (top > pr) {
		--top;
		auto x = dead[top];
		dq.push_back({{-dep[x], dp[x]}, x});	
	}
	lst = pr;
}
signed main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  int tt = 1;
  //cin >> tt;
  while (tt--) {
  	cin >> n;
  	for (int i = 1;i <= n - 1;i++) {
  		int a, b, c;
  		cin >> a >> b >> c;
  		g[a].push_back({b, c});
  		g[b].push_back({a, c});
  	}
  	for (int i = 2;i <= n;i++)
  		cin >> s[i] >> w[i];
  	dq.push_back({{0, 0}, 1});	
  	for (auto j : g[1]) {
  		dep[j.f] = j.s;
  		dfs(j.f, 1);
  	}
  	for (int i = 2;i <= n;i++)
  		cout << dp[i] << ' ';
  }
  return 0; 
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 4440 KB Output is correct
2 Correct 2 ms 4956 KB Output is correct
3 Correct 32 ms 11964 KB Output is correct
4 Correct 40 ms 15828 KB Output is correct
5 Correct 47 ms 18556 KB Output is correct
6 Correct 59 ms 21844 KB Output is correct
7 Correct 36 ms 11344 KB Output is correct
8 Execution timed out 1079 ms 14668 KB Time limit exceeded
9 Execution timed out 1037 ms 13212 KB Time limit exceeded
10 Execution timed out 1061 ms 16456 KB Time limit exceeded