Cod sursa(job #1665896)

Utilizator cordun_cristinaCristina Maria Cordun cordun_cristina Data 27 martie 2016 14:25:31
Problema Cerere Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");

const int Nmax = 100005;
int n, k[Nmax], Use[Nmax], sol[Nmax],tata[Nmax] , log2[Nmax];
vector <int> G[Nmax];

int Stramos(int q, int p)
{
    while(p--)
    {
        q = tata[q];
    }
    return q;
}

void DFS(int nod)
{
    Use[nod] = 1;
    for(int i = 0; i < (int)G[nod].size(); i++)
    {
        int vecin = G[nod][i];
        int stramos = Stramos(vecin, k[vecin]);
        if(vecin != stramos) sol[vecin] = sol[stramos] + 1;
        DFS(vecin);
    }
}

int main()
{
    f>>n;
    for(int i = 1; i <= n; i++) f>>k[i];
    for(int i = 1; i < n; i++)
    {
        int x,y;
        f>>x>>y;
        G[x].push_back(y);
        tata[y] = x;
    }
    for(int i = 1; i <= n; i++)
    {
        if(tata[i] == 0)
        {
            DFS(i);
            i = n+1;
        }
    }
    for(int i = 1; i <= n; i++)
    {
        g<<sol[i]<<' ';
    }
    return 0;
}