Cod sursa(job #2836068)

Utilizator AndreiP25Prusacov Andrei AndreiP25 Data 19 ianuarie 2022 18:39:15
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <bits/stdc++.h>
#define MAX 16002

using namespace std;

ifstream fin("asmax.in");
ofstream fout("asmax.out");

vector<int>Graf[MAX];
int S[MAX], value[MAX], viz[MAX], maxim;

void dfs(int node)
{
  viz[node]=1;
  S[node]=value[node];
  for(int i=0; i<Graf[node].size();i++)
    {
      int vecin = Graf[node][i];
      if(viz[vecin]==0)
      {
        dfs(vecin);
        if(S[vecin]>0)  
          S[node]+=S[vecin];
      }
    }
  maxim=max(maxim,S[node]);
}

int main() {
  int x,y,n;
  fin>>n;
  for(int i=1; i<=n; i++)
    {
       fin>>value[i];
      viz[i]=0;
    }
  for(int i=1; i<=n;i++)
    {
      fin>>x>>y;
      Graf[y].push_back(x);
      Graf[x].push_back(y);
    }
  dfs(1);
  fout<<maxim;
  return 0;
}