Cod sursa(job #31862)

Utilizator vlad_popaVlad Popa vlad_popa Data 16 martie 2007 18:28:59
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.39 kb
using namespace std;

#include <cstdio>
#include <cassert>
#include <string>

#define FIN "asmax.in"
#define FOUT "asmax.out"
#define NMAX 16001
#define INF 0x3f3f3f3f

int N, nf[NMAX], *f[NMAX], s[NMAX], v[NMAX], viz[NMAX], iter;

void read ()
{
  int i, x, y;

  freopen (FIN, "rt", stdin);
  freopen (FOUT, "wt", stdout);
  
  scanf ("%d\n", &N);
  for (i = 1; i <= N; ++ i)
    scanf ("%d ", &v[i]);
  scanf ("\n");
  for (i = 1; i < N; ++ i)
   {
     scanf ("%d %d\n", &x, &y);
     ++ nf[x];
     ++ nf[y];
   }
  for (i = 1; i <= N; ++ i)
   {
     f[i] = new int [nf[i] + 1];
     nf[i] = 0;
   }
   
  fseek (stdin, 0, SEEK_SET);
  
  scanf ("%d\n", &N);
  for (i = 1; i <= N; ++ i)
    scanf ("%d ", &v[i]);
  scanf ("\n");
  for (i = 1; i < N; ++ i)
   {
     scanf ("%d %d\n", &x, &y);   
     f[x][++nf[x]] = y;
     f[y][++nf[y]] = x;
   }
}

void df (int i)
{
  int j;
  bool verif = false;
  viz[i] = 1;
  s[i] = v[i];
  for (j = 1; j <= nf[i]; ++ j)
    if (viz[f[i][j]] != 1)
     {
       df (f[i][j]);
       if (s[f[i][j]] > 0)
         s[i] += s[f[i][j]];
       verif = true;
     }
  if (!verif)
    s[i] = v[i];
}

void solve ()
{
  int max = -INF;

  df(1);
  for (int i = 1; i <= N; ++ i)
    if (max < s[i])
      max = s[i];
  printf ("%d\n", max);
}

int
 main ()
{
  read ();
  solve ();
  return 0;
}