Cod sursa(job #1480195)

Utilizator ionut98Bejenariu Ionut Daniel ionut98 Data 2 septembrie 2015 11:14:35
Problema Asmax Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include<fstream>
#include<vector>
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
const int nmax=16005;
vector<int>q[nmax];
int n,val[nmax];
bool viz[nmax];
void dfs(int tata)
{
    int i,fiu;
    viz[tata]=1;
    for(i=0;i<q[tata].size();i++)
    {
        fiu=q[tata][i];
        if(viz[fiu]==0)
        {
            dfs(fiu);
            if(val[fiu]>0)
              val[tata]+=val[fiu];
        }
    }
}
int main()
{
    int i,n,x,y,sol=-10000000;
    f>>n;
    for(i=1;i<=n;i++)
      f>>val[i];
    for(i=1;i<n;i++)
    {
        f>>x>>y;
        q[x].push_back(y);
        q[y].push_back(x);
    }
    dfs(1);
    sol=val[1];
    g<<sol;
    return 0;
}