Cod sursa(job #1242277)

Utilizator alexb97Alexandru Buhai alexb97 Data 14 octombrie 2014 10:56:07
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <bitset>
using namespace std;

ifstream is("asmax.in");
ofstream os("asmax.out");

void Read();
void Dfs(int k);

int n, m;
vector<vector<int>> g;
bitset<16001> v;
vector<int> smax;

int main()
{
    Read();
    Dfs(1);
    int maxim = -0x3f3f3f3f;
    for(int i = 1; i <= n; ++i)
    {
        maxim = max(maxim, smax[i]);
    }
    os << maxim;
    is.close();
    os.close();
    return 0;
}

void Read()
{
    is >> n;
    g = vector <vector<int>>(n+1);
 //   v = vector<bool>(n + 1);
    smax = vector<int>(n + 1);
    int x, y;
    for(int i = 1; i <= n; ++i)
    {
        is >> x;
        smax[i] = x;
    }
    for(int i = 1; i <= n-1; ++i)
    {
        is >> x >> y;
        g[x].push_back(y);
        g[y].push_back(x);

    }
}

void Dfs(int k)
{
    v[k] = true;
    for (const auto& x : g[k])
    {
        if( v[x]) continue;
        Dfs(x);
        if(smax[x] > 0 )
            smax[k] += smax[x];
    }
}