Cod sursa(job #2641176)

Utilizator eugen5092eugen barbulescu eugen5092 Data 10 august 2020 13:27:57
Problema Cerere Scor 65
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include<bits/stdc++.h>
using namespace std;

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

vector<int>v[100005];
int linie[100005];
int dp[100005];
int stramos[100005];
int n,m;

void Dfs(int nod){

    if(stramos[nod]!=0){
        dp[nod]=dp[linie[m-stramos[nod]+1]]+1;
    }
    linie[++m]=nod;

    for(auto i:v[nod]){
        Dfs(i);
    }
    m--;
}

void citire(){
    ci>>n;
    int a,b;
    for(int i=1;i<=n;i++){
        ci>>stramos[i];
    }

    for(int i=1;i<=n;i++){
        ci>>a>>b;
        v[a].push_back(b);

    }
}

void rez(){
    int i;

    for(int j=1;j<=n;j++){
        if(stramos[j]==0){
            dp[j]=0;
            Dfs(j);
            break;
        }
    }
    for(i=1;i<=n;i++){
        cou<<dp[i]<<" ";
    }
}

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