Cod sursa(job #1396616)

Utilizator smallOneAdina Mateescu smallOne Data 22 martie 2015 19:22:39
Problema Cerere Scor 85
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.83 kb
#include <cstdio>
#include <iostream>
#include <fstream>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <sstream>
#include <iomanip>
#include <cmath>
#include <cstdlib>
#include <ctype.h>
#include <cstring>
#include <string>
#include <ctime>
#include <cassert>
#include <utility>

using namespace std;

#define LIM 100005
#define LG 20

int dp[LG][LIM], stramos_direct[LIM], mem[LIM], n, a, b;

int stramos(int nod, int str) {
    if(str == 0) { // stie sa rezolve problema acea maimuta (nu mai are cui sa delege)
        return 0;
    }
    int newS = nod;
    for(int i = 0; i < LG; i++) {
        if( (1 << i) & str) {
            newS = dp[i][newS];
        }
    }
    return newS;
}

int calc(int nod) {
    if(mem[nod] != -1) {
        return mem[nod];
    }
    int kstramos = stramos(nod, stramos_direct[nod]);
    if(kstramos == 0) {
        return 1;
    }
    else {
        mem[nod] = calc(kstramos) + 1;
        return mem[nod];
    }
}

int main() {
	freopen("cerere.in", "r", stdin);
	freopen("cerere.out","w", stdout);

    fill(mem, mem + LIM, -1);

    scanf("%d", &n);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &stramos_direct[i]);
    }
    for(int i = 0; i < n; i++) {
        scanf("%d %d", &a, &b);
        dp[0][b] = a;
    }

    for(int k = 1; k < LG; k++) {
        for(int i = 1; i <= n; i++) {
            dp[k][i] = dp[k - 1][dp[k - 1][i]];
        }
    }

//    for(int k = 0; k < 4; k++) {
//        for(int i = 1; i <= n; i++) {
//            cout << dp[k][i] << " ";
//        }
//        cout << "\n";
//    }

    for(int i = 1; i < n; i++) {
        printf("%d ", calc(i) - 1);
    }
    printf("%d\n", calc(n) - 1);

	return 0;
}