Cod sursa(job #3345504)

Utilizator Gabriel_DaescuDaescu Gabriel Florin Gabriel_Daescu Data 9 martie 2026 20:23:43
Problema Arbore Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <fstream>
#include <vector>
#define NMAX 100005
using namespace std;
ifstream  fin("arbore.in");
ofstream fout("arbore.out");
int N,M,cost[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);
    }
}

void update(int nod, int sum)
{
    cost[nod]+=sum;
    for(int i=0; i<tree[nod].size(); i++)
    {
        update(tree[nod][i],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();

    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;
}