Cod sursa(job #2007280)

Utilizator Andrei_CotorAndrei Cotor Andrei_Cotor Data 2 august 2017 13:49:14
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.98 kb
/*
In timpul DFS-ului folosesc o stiva a.i. stramosul al k-lea al lui nr este S[nr-k].
*/
#include<fstream>
#include<vector>
using namespace std;
ifstream fi("cerere.in");
ofstream fo("cerere.out");
int n,i,nr,x,y,K[100001],S[100001],R[100001];
vector<int> V[100001];

void dfs(int nod, int p)
{
    int i;
    nr++;
    for(i=0; i<V[nod].size(); i++)
    {
        if(V[nod][i]!=p)
        {
            if(K[V[nod][i]]!=0)
                S[nr]=S[nr-K[V[nod][i]]]+1;
            else
                S[nr]=0;
            R[V[nod][i]]=S[nr];
            dfs(V[nod][i],nod);
        }
    }
    S[nr]=0;
    nr--;
}

int main()
{
    fi>>n;
    for(i=1; i<=n; i++)
    {
        fi>>K[i];
    }
    for(i=1; i<=n; i++)
    {
        fi>>x>>y;
        V[x].push_back(y);
        V[y].push_back(x);
    }
    S[1]=0;
    nr=1;
    dfs(1,0);
    for(i=1; i<=n; i++)
        fo<<R[i]<<" ";
    fo<<"\n";
    fi.close();
    fo.close();
    return 0;
}