Cod sursa(job #1898829)

Utilizator sculap1234321Panainte Alexandru sculap1234321 Data 2 martie 2017 12:04:10
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.04 kb
#include <bits/stdc++.h>
#define val 16005
using namespace std;
vector <int> L[val];
int n, d[val];
int dp[val];
int viz[val];
void Citire()
{   int i, x, y;
    ifstream fin("asmax.in");
    fin >> n ;
    for(i = 1; i <= n; i ++)
        fin >> d[i];
    for(i = 1; i < n; i++)
    {
        fin >> x >> y;
        L[x].push_back(y);
        L[y].push_back(x);
    }


}
void DFS(int k)
{
    viz[k] = 1;
    dp[k] = d[k];

    for(int i = 0 ;i < L[k].size(); i++)
        if(viz[L[k][i]] == 0)
        {
            DFS(L[k][i]);
            if(dp[L[k][i]] > 0)
                dp[k] += dp[L[k][i]];
        }
}

int main()
{
    int i;
    Citire();
    for(i = 1; i <= n; i++)
        if(L[i].size() == 1);
            DFS(i);
    for(i = 1; i <= n; i++)
        if(viz[i] == 0)
        DFS(i);
    int maxim = dp[1];
    for(i = 2; i <= n; i++)
        maxim = max(maxim, dp[i]);
    ofstream fout("asmax.out");
    for(i = 1; i <= n; i++)
        cout<<dp[i]<<" ";
    fout << maxim;
    return 0;
}