Cod sursa(job #1075457)

Utilizator mathboyDragos-Alin Rotaru mathboy Data 8 ianuarie 2014 23:42:13
Problema Cerere Scor 100
Compilator cpp Status done
Runda Teme Pregatire ACM Unibuc 2013 Marime 1.02 kb
#include <cstdio>
#include <bitset>
#include <vector>
 
#define maxN 100005
 
using namespace std;
 
bitset <maxN> viz, mark;
int vf, N, dist[maxN], mk[maxN], st[maxN];
vector <int> g[maxN];
void df (int node)
{
    viz[node] = 1;
    for (int i = 0; i < g[node].size (); i++)
    {   
        int X = g[node][i];
        if (!viz[X])
        {   
            st[++vf] = X;
            if (mk[X])
                dist[X] = dist[st[vf - mk[X]]] + 1;
            df (g[node][i]);
            --vf;
        }
    }
}
int main ()
{
    freopen ("cerere.in", "r", stdin);
    freopen ("cerere.out", "w", stdout);
    int i, a, b;
    scanf ("%d\n", &N);
    for (i = 1; i <= N; i++) scanf ("%d", &mk[i]);
    for (i = 1; i <= N; i++)
    {
        scanf ("%d%d\n", &a, &b);
        g[a].push_back (b);
        mark[b] = 1;
    }
    for (i = 1; i <= N; i++)
        if (viz[i] == 0 && mark[i] == 0)
        {   
            st[++vf] = i;
            df (i);
        }
    for (i = 1; i <= N; i++)
        printf ("%d ", dist[i]);
    return 0;
}