Cod sursa(job #1665894)

Utilizator cordun_cristinaCristina Maria Cordun cordun_cristina Data 27 martie 2016 14:22:55
Problema Cerere Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.29 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], L[20][Nmax], log2[Nmax];
vector <int> G[Nmax];

int Stramos(int q, int p)
{
    while(p)
    {
        int k = log2[p];
        q = L[k][q];
        p = p - (1<<k);
    }
    return q;
}

void Precalculate()
{
    for(int i = 2; i <= n; i++) log2[i] = log2[i/2] + 1;
    for(int i = 1; (1<<i) <= n; i++)
        for(int j = 1; j <= n; j++) L[i][j] = L[i-1][L[i-1][j]];
}

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);
        L[0][y] = x;
    }
    Precalculate();
    for(int i = 1; i <= n; i++)
    {
        if(L[0][i] == 0)
        {
            DFS(i);
            i = n+1;
        }
    }
    for(int i = 1; i <= n; i++)
    {
        g<<sol[i]<<' ';
    }
    return 0;
}