Cod sursa(job #2507299)

Utilizator stefanpiturStefan Alexandru Pitur stefanpitur Data 9 decembrie 2019 22:07:30
Problema Asmax Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <iostream>
#include <cstdio>
using namespace std;
const int NMAX = 16000;
const int INF = 1e9;

bool visited[NMAX+1];
int last[NMAX+1],urm[NMAX+1],vf[NMAX+1],nr;
int v[NMAX+1];
int sum[NMAX+1];

void adauga(int x, int y){
    vf[++nr] = y;
    urm[nr] = last[x];
    last[x] = nr;
}

void dfs(int nod){
    visited[nod] = true;
    //sum[nod] = v[nod];
    for(int p=last[nod]; p!=0; p=urm[p]){
        int next = vf[p];
        if(!visited[next]){
            dfs(next);
            if(v[next] > 0)
                v[nod] += v[next];
        }
    }
}

int main()
{
    FILE *fin, *fout;
    int n,i,x,y,ans;
    fin = fopen("asmax.in","r");
    fout = fopen("asmax.out","w");
    fscanf(fin,"%d",&n);
    for(i=1; i<=n; i++)
        fscanf(fin,"%d",&v[i]);
    for(i=1; i<n; i++){
        fscanf(fin,"%d %d",&x,&y);
        adauga(x,y);
        adauga(y,x);
    }
    dfs(1);
    ans = v[1];
    for(i=2; i<=n; i++)
        ans = ans > v[i] ? ans : v[i];
    fprintf(fout,"%d\n",ans);
    fclose(fin);
    fclose(fout);
    return 0;
}