답안 #404638

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
404638 2021-05-14T19:09:53 Z nikatamliani Harbingers (CEOI09_harbingers) C++14
30 / 100
188 ms 36024 KB
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll N = 2e5+10, oo = 4e18;
vector<pair<int, int>> edges[N];
ll p[N], s[N], v[N], sum[N], dp[N], n;
void mini(ll &x, ll y) {
	if(x > y) {
		x = y;
	}
}
struct line {
	int k, b;
	line(int _k, int _b) : k(_k), b(_b) {}
	line() : line(-1, -1) {}
	ll f(int x) {
		if(k == -1 && b == -1) return 4e18;
		return (ll)k*x + b;
	}
};
struct node {
	node *L, *R;
	line l;
	node() : L(NULL), R(NULL), l() {}
} *root;
pair<line*, line> ops[1000005]; 
int timer = 0;
void add_line(ll l, ll r, line ln, node *&t) {
	if(t == NULL) t = new node();
	ll m = l + r >> 1;
	bool lft = t->l.f(l) < ln.f(l);
	bool mid = t->l.f(m) < ln.f(m);
	if(lft == mid && !lft) {
		line save = t->l;
		pair<line*, line> p = {&(t->l), save};
		swap(ln, t->l);
		if(timer <= 1e6)ops[++timer] = p;
	} 
	if(lft != mid && !mid) {
		line save = t->l;
		pair<line*, line> p = {&(t->l), save};
		swap(ln, t->l);
		if(timer <= 1e6)ops[++timer] = p;
	}
	if(l == r) {
		return;
	}
	if(lft == mid) {
		add_line(m+1, r, ln, t->R);
	} else {
		add_line(l, m, ln, t->L);
	}
}
ll get_min(ll l, ll r, ll x, node *&t) {
	if(t == NULL || l > x || r < x) return oo;
	if(l == r) return t->l.f(x);
	ll m = l + r >> 1;
	ll children = min(get_min(l, m, x, t->L), get_min(m+1, r, x, t->R));
	ll all = min(children, t->l.f(x));
	return all;
}
void rollback(int T) {
	while(timer > T) {
		pair<line*, line> p = ops[timer--];
		(*p.first) = p.second;
	}
}
void dfs(int x, int p) {
	int T = timer;
	if(x == 1) {
		dp[x] = 0;
	} else {
		dp[x] = get_min(1, 1e9, v[x], root) + sum[x] * v[x] + s[x];
	}
	add_line(1, 1e9, line(-sum[x], dp[x]), root);
	for(pair<int, int> to : edges[x]) {
		if(to.first != p) {
			sum[to.first] = sum[x] + to.second;
			dfs(to.first, x);
		}
	}
	rollback(T);
}
int main() {
	ios::sync_with_stdio(0); cin.tie(0);
	cin >> n;
	for(int i = 0; i < n-1; ++i) {
		int u, v, w; cin >> u >> v >> w;
		edges[u].push_back({v, w});
		edges[v].push_back({u, w});
	}
	for(int i = 2; i <= n; ++i) {
		cin >> s[i] >> v[i];
	}
	dfs(1, 0);
	for(int i = 2; i <= n; ++i) {
		cout << dp[i] << ' ';
	}
}

Compilation message

harbingers.cpp: In function 'void add_line(long long int, long long int, line, node*&)':
harbingers.cpp:30:11: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   30 |  ll m = l + r >> 1;
      |         ~~^~~
harbingers.cpp: In function 'long long int get_min(long long int, long long int, long long int, node*&)':
harbingers.cpp:57:11: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   57 |  ll m = l + r >> 1;
      |         ~~^~~
# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 20684 KB Output is correct
2 Correct 15 ms 21068 KB Output is correct
3 Incorrect 74 ms 26764 KB Output isn't correct
4 Incorrect 114 ms 30460 KB Output isn't correct
5 Runtime error 134 ms 32788 KB Memory limit exceeded
6 Runtime error 184 ms 35504 KB Memory limit exceeded
7 Correct 128 ms 27828 KB Output is correct
8 Incorrect 179 ms 31792 KB Output isn't correct
9 Runtime error 188 ms 34308 KB Memory limit exceeded
10 Runtime error 163 ms 36024 KB Memory limit exceeded