Cod sursa(job #3345505)

Utilizator Gabriel_DaescuDaescu Gabriel Florin Gabriel_Daescu Data 9 martie 2026 20:26:40
Problema Arbore Scor 15
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <fstream>
#include <vector>
#define NMAX 100005
using namespace std;
ifstream  fin("arbore.in");
ofstream fout("arbore.out");
int N,M,cost[NMAX],tata[NMAX];
vector<int> tree[NMAX];

void citire()
{
    fin>>N>>M;

    int u,v;
    for(int i=1; i<N; i++)
    {
        fin>>u>>v;
        tree[u].push_back(v);
        tree[v].push_back(u);
    }
}

void DFS(int parent, int nod)
{
    for(int i=0; i<tree[nod].size(); i++)
    {
        int next_nod=tree[nod][i];

        if(next_nod!=parent)
        {
            tata[next_nod]=nod;
            DFS(nod,next_nod);
        }
    }
}

void update(int nod, int sum)
{
    cost[nod]+=sum;
    for(int i=0; i<tree[nod].size(); i++)
    {
        int next_nod=tree[nod][i];

        if(next_nod!=tata[nod])
        {
             update(next_nod,sum);
        }
    }
}

int query(int sum)
{
    int ans=-1;
    for(int i=1; i<=N; i++)
    {
        if(cost[i]==sum)
        {
            ans=i;
            break;
        }
    }

    return ans;
}

int main()
{
    citire();

    DFS(0,1);

    int op,p,s;
    for(int q=1; q<=M; q++)
    {
        fin>>op;

        if(op==1)
        {
            fin>>p>>s;
            update(p,s);
        }

        if(op==2)
        {
            fin>>s;
            fout<< query(s) << "\n";
        }
    }

    return 0;
}