Cod sursa(job #2916447)

Utilizator AxicaVirtosu Alexandra Mihaela Axica Data 29 iulie 2022 18:39:07
Problema Cerere Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.34 kb

#include <fstream>
#include <vector>
#include <algorithm>
#define NMAX 100010
using namespace std;
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");

int n;
int nrstramos[NMAX];
int sol[NMAX];///sol[nod]=nr de cereri pt nod
vector <int> L[NMAX];///necesare pentru dfs L[nod]=fii lui nod(salvand in liste doar fii nu mai este necesar vectorul de vizitare)
int tata[NMAX];///necesar pentru a determina radacina, va fi nodul care nu are tata, adica tata[nod]=0;
int drum[NMAX];///voi tine minte lantul elementar unic de la radacina de la nodul curent
void dfs(int nod, int nivel)
{
    drum[nivel]=nod;
    sol[nod]=sol[drum[nivel-nrstramos[nod]]]+1;
    for(int i=0; i<L[nod].size(); i++)
    {
        dfs(L[nod][i], nivel+1);
    }
}
int main()
{
   FILE *fin = fopen("cerere.in","r");
    fscanf(fin,"%d",&n);
    for(int i=1; i<=n; i++)
       fscanf(fin,"%d",&nrstramos[i]);
    int x, y;
    for(int i=1; i<=n-1; i++)
    {
        fscanf(fin,"%d %d",&x,&y);
        L[x].push_back(y);
        tata[y]=x;

    }
    ///determinam radacina
    int radacina=0;
    for(int i=1; i<=n; i++)
        if(tata[i]==0)
    {
        radacina=i;
        break;
    }


    dfs(radacina, 1);
    FILE *fout = fopen("cerere.out","w");
    for(int i=1; i<=n; i++)
        fprintf(fout,"%d ",sol[i]-1);
    return 0;
}