Cod sursa(job #981398)

Utilizator raulstoinStoin Raul raulstoin Data 7 august 2013 00:23:56
Problema Arbore Scor 45
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 kb
#include<fstream>
#include<vector>

#define NMAX 100005

using namespace std;

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

int n,m;
vector<int> G[NMAX],line;
int val[NMAX],first[NMAX],last[NMAX];

void read()
{
	fin>>n>>m;
	for(int i=1,x,y;i<n;i++)
	{
		fin>>x>>y;
		G[x].push_back(y);
		G[y].push_back(x);
	}
}

void DFS(int nod,int tt)
{
	line.push_back(nod);
	first[nod]=line.size()-1;
	for(size_t i=0;i<G[nod].size();i++)
		if(G[nod][i]!=tt)
			DFS(G[nod][i],nod);
	line.push_back(nod);
	last[nod]=line.size()-1;
}

void solve()
{
	DFS(1,0);
	for(int i,aux,choice,x,y;m;m--)
	{
		fin>>choice>>x;
		if(choice==1)
		{
			fin>>y;
			val[x]+=y;
			continue;
		}
		for(i=0,aux=0;i<2*n;i++)
		{
			if(first[line[i]]==i)
				aux+=val[line[i]];
			if(aux>x)
				i=last[line[i]];
			if(last[line[i]]==i)
				aux-=val[line[i]];
			if(aux==x)
				break;
		}
		if(i==2*n)
			fout<<"-1\n";
		else
			fout<<line[i]<<'\n';
	}
}

int main()
{
	read();
	solve();
	return 0;
}