Cod sursa(job #1042971)

Utilizator alexalghisiAlghisi Alessandro Paolo alexalghisi Data 27 noiembrie 2013 20:49:03
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>

#define DN 100005
#define next_nod list[nod][i]
using namespace std;

vector<int> list[DN];
vector<int>::iterator it;
bitset<DN> viz;
int deep[DN],dp[DN],s[DN];
/// dp = numarul de stramosi in sus al unui nod prin care trebuie sa treaca
/// deep = la al catelea nod de la radacina in jos ma aflu
/// s = la al catelea nod in sus trebuie sa merg pentru a rezovla

void dfs(int nod,int niv)
{
    viz[nod]=true;
    deep[niv]=nod;
    if(s[nod])
        dp[nod]=1 + dp [ deep[ niv - s[nod] ] ];
    for(int i=0;i<list[nod].size();++i)
        if(!viz[ next_nod ])
            dfs( next_nod , niv+1);
}


int main()
{
    int n;
    ifstream f("cerere.in");
    ofstream g("cerere.out");
    f>>n;
    for(int i=1;i<=n;++i)
        f>>s[i];
    for(int i=1;i<n;++i)
    {
        int a,b;
        f>>a>>b;
        list[a].push_back(b);
    }
    dfs(1,1);
    for(int i=1;i<=n;++i)
        g<<dp[i]<<" ";
    return 0;
}