Codeforces Round #525 (Div. 2)E.（树形DP+贪心）

题目链接：http://codeforces.com/contest/1088/problem/F

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll  __int64

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))

#define pii pair<ll,ll>
#define mk(x,y) make_pair(x,y)

const int maxn =3e5+5;
const int maxm=250005;
const int mod=1e9+7;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意：选择一个k并且选择k个连通块，
要求sigma a[i]/k最大，k尽量大，
对于给定的一颗树，输出最大的分数，不用约分的形式。

题目意思需要进一步解析，
假定有一个最大的连通块，那么
根据贪心的思想，肯定是选择这个连通块，但
又要求k大，所以把所有大小为这个连通块的dp值都计数一遍，
因为我们知道两个最大的连通块不可能有交集，
如果有，肯定是交集部分就是最大值部分，其余的部分都是零，
那么我们从下往上递归求解的过程中，其实也是一种贪心选取，
在计数完当前位置的dp值后，把该位置置为无穷小防止后面再被选取。

时间复杂度：O（n）.
*/

struct node{
    int u,nxt;
}e[maxn<<1];
int head[maxn],tot=0;
void init()
{
    mst(head,-1),tot=0;
}
void add(int x,int y)
{
    e[tot]=node{y,head[x]};
    head[x]=tot++;
}

int n,a[maxn],x,y;
ll dp[maxn],ans,cnt;

void dfs(int u,int pre,int bo)
{
    dp[u]=1LL*a[u];
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].u;
        if(v==pre) continue;
        dfs(v,u,bo);
        dp[u]+=max(0LL,dp[v]);
    }
    if(bo==0) ans=max(ans,dp[u]);
    else if(dp[u]==ans) {cnt++;dp[u]=-1e15;}
}
int main()
{
    init();
    scanf("%d",&n);rep(i,1,n+1) scanf("%d",&a[i]);
    rep(i,1,n) {scanf("%d%d",&x,&y);add(x,y),add(y,x);}
    ans=-1e15,dfs(1,-1,0),dfs(1,-1,1);
    printf("%lld %lld\n",ans*cnt,cnt);
    return 0;
}