Cod sursa(job #2836424)

Utilizator AndreiP25Prusacov Andrei AndreiP25 Data 20 ianuarie 2022 12:53:08
Problema Cerere Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <bits/stdc++.h>
#define MAX 100002

using namespace std;

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

vector<int>Graf[MAX];
int viz[MAX], dp[MAX], ans[MAX], k[MAX], rootv[MAX];

void dfs(int node, int niv)
{
    viz[node]=1;
    dp[niv]=node;
    ans[node]=1+ans[dp[niv-k[node]]];
    for(int i=0; i<Graf[node].size(); i++)
    {
        if(viz[Graf[node][i]])
            dfs(i, niv+1);
    }
}



int main()
{
    int n, root;
    fin>>n;
    for(int i=1; i<=n;i++)
        fin>>k[i];
    for(int i=2; i<=n;i++)
    {
        int x,y;
        fin>>x>>y;
        Graf[x].push_back(y);
        Graf[y].push_back(x);
        rootv[y]=1;
    }
    for(int i=1; i<=n; i++)
    {
        if(rootv[i]==0)
        {
            root=i;
            break;
        }
    }
    dfs(root, 1);
    for(int i=1; i<=n;i++)
        fout<<ans[i];
    return 0;
}