Cod sursa(job #2205590)

Utilizator nicolaefilatNicolae Filat nicolaefilat Data 19 mai 2018 15:54:51
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <iostream>
#include <fstream>
#include <vector>
#define MAXN 100005

using namespace std;

ifstream in("cerere.in");
ofstream out("cerere.out");

vector<int>graf[MAXN];
bool viz[MAXN],e_sursa[MAXN];
int n,stramosi[MAXN],drum[MAXN],dp[MAXN],sursa,nr_noduri;

void dfs(int nod){
    drum[++nr_noduri] = nod;
    if(stramosi[nod])
        dp[nod] = dp[drum[nr_noduri - stramosi[nod]]] + 1;
    viz[nod] = true;
    int curent;
    for(int i = 0; i < graf[nod].size(); i++){
        curent = graf[nod][i];
        if(!viz[curent])
            dfs(curent);

    }
    --nr_noduri;
}

int main()
{
    in>>n;
    for(int i = 1; i <= n; i++)
        in>>stramosi[i];
    for(int i = 1; i < n; i++){
        int tata,fiu;
        in>>tata>>fiu;
        e_sursa[fiu] = true;
        graf[tata].push_back(fiu);
    }
    for(int i = 1; i <= n && !sursa; i++)
        if(!e_sursa[i])
            sursa = true;
    dfs(sursa);
    for(int i = 1; i <= n; i++)
        out<<dp[i]<<" ";
    return 0;
}