Cod sursa(job #1502041)

Utilizator patrutoiuandreipatrutoiu andrei patrutoiuandrei Data 14 octombrie 2015 06:40:02
Problema Cerere Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <fstream>
#include <string.h>
#include <vector>

#define dim 100010
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,m,i,j,k,nr,a,v[dim],sol[dim],val[dim];
char s[7*dim];
vector <int> G[dim];
void DFS(int x,int niv)
{
    if(v[x]==0)
    {
        sol[x]=0;
        val[niv]=0;
    }
    else
    {
        sol[x]=val[niv-v[x]]+1;
        val[niv]=sol[x];
    }
    for(int i=0;i<G[x].size();i++)
    {
        DFS(G[x][i],niv+1);
    }
}
int main()
{
    fin>>n;
    fin.get();
    fin.getline(s+1,700010);
    m=strlen(s+1);
    for(i=1;i<=m;i++)
    {
        if(s[i]>='0'&&s[i]<='9')
        {
            nr=nr*10+s[i]-'0';
        }
        else
            v[++k]=nr,nr=0;
    }
    v[++k]=nr;
    for(i=1;i<=n;i++)
    {
        fin.getline(s+1,20);
        m=strlen(s+1);
        nr=0;
        a=0;
        for(j=1;j<=m;j++)
        {
            if(s[j]>='0'&&s[j]<='9')
            {
                nr=nr*10+s[j]-'0';
            }
            else
            {
                a=nr;
                nr=0;
            }
        }
        G[a].push_back(nr);
    }
    DFS(1,1);
    for(i=1;i<=n;i++)
        fout<<sol[i]<<' ';
    return 0;
}