zoukankan      html  css  js  c++  java
  • Codeforces 743D Chloe and pleasant prizes(树型DP)

                                                                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


    题目就是让你找出两棵不相交的子树使得这两棵子树结点权值和最大。
    那么对于每个结点,如果他有大于等于两个儿子,那么就计算出他的后代中权值最大的两棵不相交的子树,
    这两棵子树的信息传递到他的几个儿子上。(几个儿子中挑两个最大的)
    那么最后总的贪心一下排个序统计一下就好了。
    具体细节可以看代码。


     1 #include <bits/stdc++.h>
     2 
     3 using namespace std;
     4 
     5 #define rep(i,a,b)              for(int i(a); i <= (b); ++i)
     6 #define for_edge(i,x)           for(int i = H[x]; i; i = X[i])
     7 #define LL              long long
     8 #define PB              push_back
     9 #define sz(x)            (int)v[x].size()
    10 
    11 const int N     =    200000      +       10;
    12 
    13 vector <LL> v[N];
    14 
    15 LL sum[N];
    16 LL a[N];
    17 int E[N << 2], H[N << 2], X[N << 2];
    18 int x, y, n;
    19 int et = 0;
    20 LL ret[N];
    21 LL c[N];
    22 
    23 inline void addedge(int a, int b){
    24     E[++et] = b, X[et] = H[a], H[a] = et;
    25     E[++et] = a, X[et] = H[b], H[b] = et;
    26 }
    27 
    28 
    29 void dfs(int x, int fa){
    30     sum[x] = a[x];
    31     for_edge(i, x){
    32         int now = E[i];
    33         if (now != fa){
    34             dfs(now, x);
    35             sum[x] += sum[now];
    36         }
    37     }
    38 }
    39             
    40 void dfs2(int x, int fa){
    41     ret[x] = sum[x];
    42     for_edge(i, x){
    43         int now = E[i];
    44         if (now == fa) continue;
    45         dfs2(now, x);
    46         v[x].PB(ret[now]);
    47         ret[x] = max(ret[x], ret[now]);
    48     }
    49 }
    50 
    51 bool cmp(LL a, LL b){ return a > b;}
    52 
    53 int main(){
    54 
    55     scanf("%d", &n);
    56     rep(i, 1, n) scanf("%lld", &a[i]);
    57 
    58     rep(i, 1, n - 1){
    59         scanf("%d%d", &x, &y);
    60         addedge(x, y);
    61     }
    62 
    63     dfs(1, 0);
    64     dfs2(1, 0);
    65     
    66     LL ans = -1000000000000000000LL;
    67 
    68     rep(i, 1, n) if (sz(i) > 1) sort(v[i].begin(), v[i].end(), cmp);
    69     rep(i, 1, n) if (sz(i) > 1) ans = max(ans, v[i][0] + v[i][1]);
    70 
    71     if (ans != -1000000000000000000LL) printf("%lld
    ", ans);
    72     else puts("Impossible");
    73 
    74 
    75     return 0;
    76 
    77 }
    
    
  • 相关阅读:
    kvm初体验之四:从Host登录Guest的五种方式
    kvm初体验之三:vm的安装及管理
    CentOS Wifi Connection
    kvm初体验之二:安装
    kvm初体验之一:参考文档
    有6种不同颜色的球,每种球有无数个。现在取5个球,求取出5、4、3、2种不同颜色球的概率分别为多少
    求两个字符串的最长连续子串
    不用除法来实现整数的除法运算
    抽象类和接口的区别
    o(n)的时间复杂判断回文数
  • 原文地址:https://www.cnblogs.com/cxhscst2/p/6416899.html
Copyright © 2011-2022 走看看