Cod sursa(job #1398199)

Utilizator smallOneAdina Mateescu smallOne Data 24 martie 2015 01:22:07
Problema Cerere Scor 55
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.71 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 stk[LIM], res[LIM], stramos_direct[LIM], n, a, b, radacina, szSTK;
vector<int> arr[LIM];
bool son[LIM], viz[LIM];

void calc(int nod) {
    stk[szSTK++] = nod;
    viz[nod] = true;
    int pos = szSTK - stramos_direct[nod] - 1;
    if(pos >= 0) {
        res[nod] = res[stk[pos]] + 1;
    }
    for(int i = 0; i < (int) arr[nod].size(); i++) {
        int vec = arr[nod][i];
        if(!viz[vec]) {
            calc(vec);
            szSTK--;
        }
    }
}

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

    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);
        arr[a].push_back(b);
        arr[b].push_back(a);
        son[b] = true;
    }

    for(int i = 1; i <= n; i++) {
        if(son[i] == false) {
            radacina = i;
        }
    }

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

    szSTK = 0;
    calc(radacina);
    for(int i = 1; i < n; i++) {
        printf("%d ", res[i] - 1);
    }
    printf("%d\n", res[n] - 1);

	return 0;
}