Cod sursa(job #2490130)

Utilizator NashikAndrei Feodorov Nashik Data 9 noiembrie 2019 19:38:28
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.46 kb
//#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
queue <int> q;
vector<int> v[16005];
int d1[16005],d2[16005],cost[16005],tata[16005],f[16005],maxim=-1,maxi,fol[16005];
int main()
{
    ifstream cin("asmax.in");
    ofstream cout("asmax.out");
    int n,a,b;
    cin>>n;
    cin>>cost[1];
    maxi=cost[1];
    for(int i=2;i<=n;i++){
        cin>>cost[i];
        if(cost[i]>maxi)
            maxi=cost[i];
    }
    if(maxi<0){
        cout<<maxi;
        return 0;
    }
    for(int i=1;i<n;i++){
        cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
        f[a]++;
        f[b]++;
    }
    for(int i=1;i<=n;i++){
        if(f[i]==1)
            q.push(i);
    }
    int x;
    while(q.empty()==false){
        x=q.front();
        q.pop();
        int sum=0;
        maxim=-1;
        tata[x]=-1;
        for(int i=0;i<v[x].size();i++){
            if(fol[v[x][i]]==true){
                sum+=d1[v[x][i]];
                if(maxim<d2[v[x][i]])
                    maxim=d2[v[x][i]];
            }
            else
                tata[x]=v[x][i];
        }
        sum+=cost[x];
        if(sum<0){
            sum=0;
        }
        d1[x]=sum;
        d2[x]=max(maxim,sum);
        ///verific ca sa nu scad tatal radacinei
        if(tata[x]<=n and tata[x]>=1){
            f[tata[x]]--;
            if(f[tata[x]]<=1)
                q.push(tata[x]);
        }
        fol[x]=true;
    }
    cout<<d2[x];
    return 0;
}