Cod sursa(job #2954401)

Utilizator TudosieRazvanTudosie Marius-Razvan TudosieRazvan Data 14 decembrie 2022 10:12:21
Problema Asmax Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <queue>
#include <climits>
#define NMAX 16003
using namespace std;

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

int n,m,start;
long long dp[NMAX];
int cost[NMAX];
bool viz[NMAX];
vector<int>graf[NMAX];

long long dfs(int nod)
{
    viz[nod]=true;
    dp[nod]=cost[nod];
    long long adun=0;
    for(int i=0; i<graf[nod].size(); i++)
    {
        int nd=graf[nod][i];
        if(!viz[nd])
        {
            adun=max(adun,adun+dfs(nd));
        }
    }
    dp[nod]=dp[nod]+adun;
    return dp[nod];
}

int main()
{
    fin>>n;
    for(int i=1; i<=n; i++)
    {
        fin>>cost[i];
    }
    int x,y;
    while(fin>>x>>y)
    {

        graf[x].push_back(y);
        graf[y].push_back(x);

    }
    fout<<dfs(1);

    return 0;
}