Pagini recente » Arhiva de probleme | Cod sursa (job #2483974) | Cod sursa (job #2502317) | Cod sursa (job #574088) | Cod sursa (job #970737)
Cod sursa(job #970737)
#include <fstream>
#include <vector>
#define Nmax 16099
#define Vizitat 1
#define Nevizitat 0
#define Infinit 1<<30
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
int n,v[Nmax],C[Nmax],S;
vector <int> a[Nmax];
char viz[Nmax];
void ReadInput()
{
f>>n;
for (int i=1; i<=n; i++) f>>v[i];
for (int i=1; i<=n-1; i++)
{
int x,y;
f>>x>>y;
a[x].push_back(y); a[y].push_back(x);
}
}
void dfs(int nod)
{
viz[nod]=Vizitat; C[nod]=v[nod];
for (int i=0; i<a[nod].size();i++)
if (viz[a[nod][i]]==Nevizitat)
{
dfs(a[nod][i]);
if (C[a[nod][i]]>0)C[nod]+=C[a[nod][i]];
}
if (C[nod]>S) S=C[nod];
}
void Solve()
{
S=-Infinit;
dfs(1);
g<<S<<'\n';
}
int main()
{
ReadInput();
Solve();
f.close();g.close();
return 0;
}