B. 齐心抗疫

B. 齐心抗疫

 

 

 题解：

 AC_Code:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std;
typedef long long ll;
const int maxn = 5e4+10;
const int inf = 0x3f3f3f3f;
const int mod = 998244353;
#define rep(i,first,second) for(int i=first;i<=second;i++)
#define dep(i,first,second) for(int i=first;i>=second;i--)

struct edge{int to,nxt;}e[maxn<<1];
int head[maxn<<1],tot;
int n,a[maxn];
int maxdis,maxv;
int d1[maxn],d2[maxn];

void add(int u,int v){
    e[tot].to = v;
    e[tot].nxt = head[u];
    head[u] = tot++;
}

void dfs(int u,int f,int dis,int d[]){
    d[u]=dis;
    if( d[u]>d[maxv] ){
        maxv = u;
    }

    for(int i=head[u];~i;i=e[i].nxt){
        int v=e[i].to;
        if( v==f ) continue;
        dfs(v,u,dis+1,d);
    }
}


int main()
{
    memset(head,-1,sizeof(head));

    scanf("%d",&n);
    rep(i,1,n) scanf("%d",&a[i]);

    rep(i,1,n-1){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }

    int s=1;
    d1[maxv]=0;
    dfs(1,-1,0,d1);

    s=maxv,d1[maxv]=0;
    dfs(s,-1,0,d1);

    s=maxv,d2[maxv]=0;
    dfs(s,-1,0,d2);

    int maxx=-1;
    rep(i,1,n){
        maxx = max(maxx,max(d1[i],d2[i])*a[i]);
    }
    printf("%d
",maxx);

    return 0;
}