Cod sursa(job #2046738)

Utilizator vladboss2323Ciorica Vlad vladboss2323 Data 24 octombrie 2017 08:45:08
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.19 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;

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

const int N=100001;
vector <int> a[N];
int n,v[N],nr,m,rez;
bool viz[N];
void citire()
{
    int x,y,i;
    in>>n;
    for(i=1; i<=n; i++)
        in>>v[i];
    for(i=1; i<n; i++)
    {
        in>>x>>y;
        a[y].push_back(x);
    }
    in.close();
}

void dfs(int nod)
{
    int y;
    viz[nod]=1;
    for(size_t i=0; i<a[nod].size(); i++)
    {
        y=a[nod][i];
        if(!viz[y])
        {
            nr++;
            if(y==1)
            {
                rez++;
                return;
            }
            if(nr%m==0)
            {
                rez++;
                if(v[y]==0)
                    return;
            }
            dfs(y);
        }
    }
}

int main()
{
    citire();
    int i;
    for(i=1; i<=n; i++)
    {

        nr=0;
        rez=0;
        if(v[i]==0)
            out<<v[i]<<" ";
        else
        {

            m=v[i];
            dfs(i);
            out<<rez<<" ";
        }
        for(int j=1; j<=n; j++)
            viz[j]=0;
    }
    return 0;
}