Cod sursa(job #2138970)

Utilizator luanastLuana Strimbeanu luanast Data 21 februarie 2018 23:32:05
Problema Cerere Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");
vector <int> L[10010];
int n,i,x,y,f[100010],v[100010],k,K[100010],sol[100010],rad;
//simulezi intr-un vector dfs
//            1
//      2           5
//  3       4           6
//dfs de duce i 2 3 apoi scoate pe 3 si merge in 4 (strmosii tot 1 si 2 raman) etc
//deci sol i este sol elementului al Ki-lea de la coada
//sol se calculeaza de la rad spre frunze deci sol de stramos de i e deja calculat!
void dfs(int nod){

    if(K[nod]>0)
        sol[nod]=1+sol[v[k-K[nod]+1]];
    v[++k]=nod;

    for(int i=0;i<L[nod].size();i++)
        dfs(L[nod][i]);
    --k;
}

int main(){
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>K[i];
    for(i=1;i<n;i++){
        fin>>x>>y;
        L[x].push_back(y);
        f[y]=1;
    }
    for(i=1;i<=n;i++)
        if(f[i]==0){
            rad=i;
            break;
        }
    dfs(rad);
    for(i=1;i<=n;i++)
        fout<<sol[i]<<" ";
}