Cod sursa(job #1507007)

Utilizator cristibogdanPatrascu Cristian cristibogdan Data 21 octombrie 2015 10:47:34
Problema Cerere Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.22 kb
#include <fstream>
#include <vector>
#include <string.h>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int VIZ[100001],t[100001],v[100001],i,j,x,y,n,k,cmi,T;
bool viz[100001],vviz[100001];
char sir[10001];
vector <int> G[100001];
      void DFS(int nod){
    viz[nod]=1;
   if(v[nod]==0)
    {
        VIZ[nod]=0;
        t[k]=0;
    }
    else
    {
        VIZ[nod]=t[k-v[nod]]+1;
        t[k]=VIZ[nod];}
        for(int i=0;i<G[nod].size();i++)
                if(viz[G[nod][i]]==0){
                    k++;
                    DFS(G[nod][i]);
k--;
                    }

}
int main()
{
 f>>n;

 for(i=1;i<=n;i++)
        f>>v[i];
        f.get();
for(j=1;j<=n-1;j++){
        f.getline(sir+1, 1000);
        T=strlen(sir+1);
        for(i=1;i<=T;i++)
        {
            if(sir[i]>='0'&&sir[i]<='9')
                y=y*10+sir[i]-'0';
            else
            {
                x=y;
                y=0;
            }
        }

        G[x].push_back(y);
        vviz[y]=1;
        y=0;
    }

    for(i=1;i<=n;i++)
    if(vviz[i]==0){
        cmi=i;
        break;
    }
DFS(cmi);
    for(i=1;i<=n;i++)
            g<<VIZ[i]<<" ";

    return 0;
}