Cod sursa(job #1000723)

Utilizator Adrian1997Radulescu Adrian Adrian1997 Data 23 septembrie 2013 17:30:16
Problema Cerere Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n,S[100011],tr[100011];
bool viz[100011];

vector<int> L[100011];

void dfs(int nod){
    S[++S[0]]=nod;
    int q=tr[nod];
    tr[nod]=S[S[0]-q];
    for(register int i=0;i<L[nod].size();i++){
        dfs(L[nod][i]);
    }
    S[0]--;
}

int main(void){
    register int i,j,x,y;

    f>>n;
    for(i=1;i<=n;i++)
        f>>tr[i];
    for(i=1;i<n;i++){
        f>>x>>y;
        viz[y]=true;
        L[x].push_back(y);
    }

    for(i=1;i<=n;i++)
        if(!viz[i]){
            dfs(i);
            break;
        }
    for(i=1;i<=n;i++){
        y=0,x=i;
        while(tr[x]!=x){
            x=tr[x];
            y++;
        }
        g<<y<<" ";
    }
    f.close();
    g.close();
    return 0;
}