Cod sursa(job #30984)

Utilizator vlad_popaVlad Popa vlad_popa Data 15 martie 2007 13:12:56
Problema Asmax Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.45 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];
int *fi[NMAX], nfi[NMAX], viz[NMAX], iter;

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

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

void df (int i)
{
  int j;
  bool verif = false;
  viz[i] = iter;
  s[i] = v[i];
  for (j = 1; j <= nf[i]; ++ j)
    if (viz[f[i][j]] != iter)
     {
       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;

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

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