Codeforces Round #384 (Div. 2)-D. Chloe and pleasant prizes(线段树)
2016-12-16 11:23
393 查看
原题链接
D. Chloe and pleasant prizes
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
Generous sponsors of the olympiad in which Chloe and Vladik took part allowed all the participants to choose a prize for them on their own. Christmas is coming, so sponsors decided to decorate the Christmas tree with their prizes.
They took n prizes for the contestants and wrote on each of them a unique id (integer from 1 to n).
A gift i is characterized by integer ai —
pleasantness of the gift. The pleasantness of the gift can be positive, negative or zero. Sponsors placed the gift 1 on the top of the tree. All the other gifts
hung on a rope tied to some other gift so that each gift hung on the first gift, possibly with a sequence of ropes and another gifts. Formally, the gifts formed a rooted tree with n vertices.
The prize-giving procedure goes in the following way: the participants come to the tree one after another, choose any of the remaining gifts and cut the rope this prize hang on. Note that all the ropes which were used to hang other
prizes on the chosen one are not cut. So the contestant gets the chosen gift as well as the all the gifts that hang on it, possibly with a sequence of ropes and another gifts.
Our friends, Chloe and Vladik, shared the first place on the olympiad and they will choose prizes at the same time! To keep themselves from fighting, they decided to choose two different gifts so that the sets of the gifts that
hang on them with a sequence of ropes and another gifts don't intersect. In other words, there shouldn't be any gift that hang both on the gift chosen by Chloe and on the gift chosen by Vladik. From all of the possible variants they will choose such pair of
prizes that the sum of pleasantness of all the gifts that they will take after cutting the ropes is as large as possible.
Print the maximum sum of pleasantness that Vladik and Chloe can get. If it is impossible for them to choose the gifts without fighting, print Impossible.
Input
The first line contains a single integer n (1 ≤ n ≤ 2·105) —
the number of gifts.
The next line contains n integers a1, a2, ..., an ( - 109 ≤ ai ≤ 109) —
the pleasantness of the gifts.
The next (n - 1) lines contain two numbers each. The i-th
of these lines contains integers ui and vi (1 ≤ ui, vi ≤ n, ui ≠ vi) —
the description of the tree's edges. It means that gifts with numbers ui and vi are
connected to each other with a rope. The gifts' ids in the description of the ropes can be given in arbirtary order: vi hangs
on ui or ui hangs
on vi.
It is guaranteed that all the gifts hang on the first gift, possibly with a sequence of ropes and another gifts.
Output
If it is possible for Chloe and Vladik to choose prizes without fighting, print single integer — the maximum possible sum of pleasantness they can get together.
Otherwise print Impossible.
Examples
input
output
input
output
input
output
对树进行深搜产生dfs序列,在dfs序列上建立线段树,线段树的节点表示范围内的最大值。
再次对树进行深搜对于每个节点,把他的所有祖先节点的值和子孙节点的值在线段树上减去INF,此时求出最大值p,再加上该节点的值更新ans
D. Chloe and pleasant prizes
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
Generous sponsors of the olympiad in which Chloe and Vladik took part allowed all the participants to choose a prize for them on their own. Christmas is coming, so sponsors decided to decorate the Christmas tree with their prizes.
They took n prizes for the contestants and wrote on each of them a unique id (integer from 1 to n).
A gift i is characterized by integer ai —
pleasantness of the gift. The pleasantness of the gift can be positive, negative or zero. Sponsors placed the gift 1 on the top of the tree. All the other gifts
hung on a rope tied to some other gift so that each gift hung on the first gift, possibly with a sequence of ropes and another gifts. Formally, the gifts formed a rooted tree with n vertices.
The prize-giving procedure goes in the following way: the participants come to the tree one after another, choose any of the remaining gifts and cut the rope this prize hang on. Note that all the ropes which were used to hang other
prizes on the chosen one are not cut. So the contestant gets the chosen gift as well as the all the gifts that hang on it, possibly with a sequence of ropes and another gifts.
Our friends, Chloe and Vladik, shared the first place on the olympiad and they will choose prizes at the same time! To keep themselves from fighting, they decided to choose two different gifts so that the sets of the gifts that
hang on them with a sequence of ropes and another gifts don't intersect. In other words, there shouldn't be any gift that hang both on the gift chosen by Chloe and on the gift chosen by Vladik. From all of the possible variants they will choose such pair of
prizes that the sum of pleasantness of all the gifts that they will take after cutting the ropes is as large as possible.
Print the maximum sum of pleasantness that Vladik and Chloe can get. If it is impossible for them to choose the gifts without fighting, print Impossible.
Input
The first line contains a single integer n (1 ≤ n ≤ 2·105) —
the number of gifts.
The next line contains n integers a1, a2, ..., an ( - 109 ≤ ai ≤ 109) —
the pleasantness of the gifts.
The next (n - 1) lines contain two numbers each. The i-th
of these lines contains integers ui and vi (1 ≤ ui, vi ≤ n, ui ≠ vi) —
the description of the tree's edges. It means that gifts with numbers ui and vi are
connected to each other with a rope. The gifts' ids in the description of the ropes can be given in arbirtary order: vi hangs
on ui or ui hangs
on vi.
It is guaranteed that all the gifts hang on the first gift, possibly with a sequence of ropes and another gifts.
Output
If it is possible for Chloe and Vladik to choose prizes without fighting, print single integer — the maximum possible sum of pleasantness they can get together.
Otherwise print Impossible.
Examples
input
8 0 5 -1 4 3 2 6 5 1 2 2 4 2 5 1 3 3 6 6 7 6 8
output
25
input
4 1 -5 1 1 1 2 1 4 2 3
output
2
input
1 -1
output
Impossible
对树进行深搜产生dfs序列,在dfs序列上建立线段树,线段树的节点表示范围内的最大值。
再次对树进行深搜对于每个节点,把他的所有祖先节点的值和子孙节点的值在线段树上减去INF,此时求出最大值p,再加上该节点的值更新ans
#include <bits/stdc++.h> #define maxn 200005 #define INF 2e15 typedef long long ll; using namespace std; struct Node{ int l, r; }node[maxn]; int p[maxn], cnt, sum[maxn], n, e; ll d[maxn], num[maxn], T[maxn<<2], ans = -INF; vector<int> v[maxn]; void dfs1(int j, int f){ p[++cnt] = j; node[j].l = cnt; sum[j] = 1; for(int i = 0; i < v[j].size(); i++){ int h = v[j][i]; if(h != f){ dfs1(h, j); d[j] += d[h]; sum[j] += sum[h]; } } d[j] += num[j]; node[j].r = cnt; } void Build(int s, int l, int r){ if(l == r){ T[s] = d[p[l]]; return ; } int mid = (l + r) >> 1; Build(s<<1, l, mid); Build(s<<1|1, mid+1, r); T[s] = max(T[s<<1], T[s<<1|1]); } void Update(int s, int l, int r, int L, int R, ll c){ if(l == L && R == r){ T[s] += c; return ; } int mid = (L + R) >> 1; if(r <= mid) Update(s<<1, l, r, L, mid, c); else if(l > mid) Update(s<<1|1, l, r, mid+1, R, c); else{ Update(s<<1, l, mid, L, mid, c); Update(s<<1|1, mid+1, r, mid+1, R, c); } T[s] = max(T[s<<1], T[s<<1|1]); } void dfs2(int j, int f, int h){ ++e; int kk = e; Update(1, kk, kk, 1, n, (ll)-INF); for(int i = 0; i < v[j].size(); i++){ int s = v[j][i]; if(s != f){ if(h + sum[s] != n){ Update(1, node[s].l, node[s].r, 1, n, (ll)-INF); ans = max(ans, d[s] + T[1]); Update(1, node[s].l, node[s].r, 1, n, (ll)INF); } dfs2(s, j, h+1); } } Update(1, kk, kk, 1, n, (ll)INF); } int main(){ // freopen("in.txt", "r", stdin); int a, b; scanf("%d", &n); for(int i = 1; i <= n; i++) scanf("%I64d", num+i); for(int i = 1; i < n; i++){ scanf("%d%d", &a, &b); v[a].push_back(b); v[b].push_back(a); } dfs1(1, -1); Build(1, 1, n); dfs2(1, -1, 1); if(ans == -INF) puts("Impossible"); else printf("%I64d\n", ans); return 0; }
相关文章推荐
- Codeforces Round #384(Div. 2)D. Chloe and pleasant prizes【树形dp】
- Codeforces Round #384 (Div. 2)D. Chloe and pleasant prizes(树DP)
- Codeforces Round #384 (Div. 2) D. Chloe and pleasant prizes(树形DP)
- CodeForces743D - Chloe and pleasant prizes(树型DP)
- Codeforces Round #384 (Div. 2)D - Chloe and pleasant prizes 树形dp
- coderforces #384 D Chloe and pleasant prizes(DP)
- CodeForces-743D Chloe and pleasant prizes(树型dp)
- Codeforces Round #384 (Div. 2)D. Chloe and pleasant prizes
- Codeforces Round #384 (Div. 2)D - Chloe and pleasant prizes 树形dp
- 【Codeforces 743D】 Chloe and pleasant prizes【树形dp】
- Codeforces Round #384 (Div. 2) D. Chloe and pleasant prizes
- codeforces 743 D. Chloe and pleasant prizes(DFS)
- Codeforces Round #337 (Div. 2) D. Vika and Segments 线段树 矩阵面积并
- Codeforces Round #343 (Div. 2)D - Babaei and Birthday Cake 线段树
- Codeforces Round #384 (Div. 2)B. Chloe and the sequence 数学
- 有趣的线段树小集合 Codeforces Round #250 (Div. 1) D. The Child and Sequence
- Codeforces Round #292 (Div. 1) C. Drazil and Park 线段树
- Codeforces Round #384 (Div. 2) B. Chloe and the sequence
- Codeforces Round #384 (Div. 2) B. Chloe and the sequence(规律题)
- Codeforces Round #442 (Div. 2) E. Danil and a Part-time Job (dfs序树型转线性 线段树区间修改区间查询)