Cod sursa(job #1316571)

Utilizator addy01adrian dumitrache addy01 Data 13 ianuarie 2015 21:59:18
Problema Cerere Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <iostream>
#include <fstream>
#include <vector>
#define maxn 100010
using namespace std;

ifstream in("cerere.in");
ofstream out("cerere.out");

int n,nr[maxn],tata[maxn];
bool viz[maxn];
vector <int> Graf[maxn];
int sol[maxn];

void DFS(int nod)
{
    viz[nod]=1;
    for(auto i : Graf[nod])
        if(!viz[i])
        {
            DFS(i);
        }
        int x=nod;
            while(nr[x])
                {
                    int z=x;
                    for(int y=1;y<=nr[z];y++)
                        x=tata[x];
                    sol[nod]++;
                }
}

int main()
{
    int i,x,y;
    in>>n;
    for(i=1;i<=n;i++)
        in>>nr[i];

    while(in>>x>>y)
    {
        Graf[x].push_back(y);
        Graf[y].push_back(x);
        tata[y]=x;
    }

    for(i=1;i<=n;i++)
        if(!viz[i])
            DFS(i);


    for(i=1;i<=n;i++)
        out<<sol[i]<<" ";
    return 0;
}