Cod sursa(job #2144292)

Utilizator DovlecelBostan Andrei Dovlecel Data 26 februarie 2018 17:38:25
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
vector <int> a[100001];
int n,m,v[100001],rasp[100001],st[100001],vf,t[100001];
bool viz[100001];
ifstream in("cerere.in");
ofstream out("cerere.out");
void dfs(int x)
{
    st[++vf]=x;
    if(v[x]==0)
        rasp[x]=0;
    else
        rasp[x]=rasp[st[vf-v[x]]]+1;
    int y;
    viz[x]=true;
    for(unsigned i=0;i<a[x].size();i++)
    {
        y=a[x][i];
        if(!viz[y])
            dfs(y);
    }
    vf--;
}
int main()
{
    int x,y,j,i;
    in>>n;
    for(j=1;j<=n;j++)
        in>>v[j];
    for(j=1;j<n;j++)
    {
        in>>x>>y;
        t[y]=x;
        a[x].push_back(y);
    }
    for(i=1;i<=n;i++)
        if(t[i]==0)
        {
            dfs(i);
            break;
        }
    for(j=1;j<=n;j++)
       out<<rasp[j]<<' ';
    return 0;
}