Cod sursa(job #1526679)

Utilizator ducu34Albastroiu Radu Gabriel ducu34 Data 16 noiembrie 2015 23:57:04
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.17 kb
#include <fstream>
#include <algorithm>
#include <vector>
#define MIN -2147480000

using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int MAX = MIN,i,n,x,y;
vector<int>v[16001];

struct arb
{
    int nod,subarbore;
}valori[16001];

void f(int k, int t)
{
    int i,x,y,val,z;
    for(i=0;i<v[k].size();i++)
    {
        val = v[k][i];
        if(val != t)
        {
            f(val, k);
            x = max(valori[k].nod+valori[val].nod, valori[k].subarbore+valori[val].nod);
            y = max(valori[k].nod+valori[val].subarbore, valori[k].subarbore+valori[val].subarbore);
            z = max(x, y);
            valori[k].subarbore = max(z, valori[k].subarbore);
            MAX = max(MAX, valori[k].subarbore);
        }
    }
}

int main()
{
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>valori[i].nod;
    for(i=1;i<n;i++)
    {
        fin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    for(i=1;i<=n;i++)
        valori[i].subarbore = valori[i].nod;
    f(1,0);
    for(i=1;i<=n;i++)
        if(valori[i].subarbore>MAX)
            MAX=valori[i].subarbore;
    fout<<MAX;
    return 0;
}