Cod sursa(job #2014508)

Utilizator LizaSzabo Liza Liza Data 23 august 2017 19:50:57
Problema Arbore Scor 25
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.16 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("arbore.in");
ofstream fout("arbore.out");
const int NMax=100005;
vector <int> G[NMax];
int N,M,SUM[NMax],F[NMax],sol,k,IN[NMax];

 void Read()
 {
     fin>>N>>M;
     for(int i=1;i<=N-1;++i)
     {
         int x,y;
         fin>>x>>y;
         F[y]=x;
         G[x].push_back(y);
         G[y].push_back(x);
     }
     F[1]=0;

 }


 void DFS(int Nod, int Father,int s)
 {
    SUM[Nod]=SUM[Father]+IN[Nod];
     if(SUM[Nod]==s)
     {
         sol=Nod;
     }
     for(unsigned int j=0;j<G[Nod].size();++j)
     {
         int Vecin=G[Nod][j];
         if(Vecin!=Father)
         {
             DFS(Vecin,Nod,s);
         }
     }
 }

void Solve1()
{
    int p,s;
    fin>>p>>s;
    IN[p]+=s;
}
void Solve2()
{
    int s;
    fin>>s;
    sol=-1;
    DFS(1,0,s);
    fout<<sol<<"\n";
}

int main()
{
     Read();
    for(int i=1;i<=M;++i)
     {
         int x;
         fin>>x;
        if(x==1)
        {
            Solve1();
        }
        else
        {
            Solve2();
        }
     }
    return 0;
}