Cod sursa(job #1481491)

Utilizator ArchazeyBaltatu Andrei-Mircea Archazey Data 4 septembrie 2015 17:35:02
Problema Arbore Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.1 kb
#include<bits/stdc++.h>
using namespace std;

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

const int MODULO=10007;
const int NMAX=100005;
const int SQMAX=405;

int n,m,len,sq,eu[NMAX],start[NMAX],stop[NMAX],which[NMAX];
int add[SQMAX],a[NMAX];
vector<int>v[NMAX];
map<int,int>M[SQMAX];

inline void Dfs(int x,const int f)
{
    eu[++len]=x;
    start[x]=len;
    for (vector<int>::iterator it=v[x].begin();it!=v[x].end();it++)
        if ((*it)!=f)
            Dfs(*it,x);
    stop[x]=len;
}

inline void Solve(int x,int y,int who,int val)
{
    int i,st,dr;
    M[who].clear();
    for (i=x;i<=y;i++) a[i]+=val;
    dr=min(n,who*sq);st=(who-1)*sq+1;
    for (i=st;i<=dr;i++)
        M[who][a[i]]=eu[i];
}

int main()
{
    int i,j,l,x,y,s,op,aux,poz;
    fin>>n>>m;sq=sqrt(n);
    for (i=1;i<n;i++)
        {
            fin>>x>>y;
            v[x].push_back(y);
            v[y].push_back(x);
        }
    Dfs(1,0);
    for (i=1;i<=n;i++)
        {
            which[i]=(i/sq)+1;
            if (i%sq==0) which[i]--;
            if (which[i]!=which[i-1]) M[which[i]][0]=eu[i];
        }
    for (l=1;l<=m;l++)
        {
            fin>>op;
            if (op==1)
                {
                    fin>>x>>s;
                    i=start[x];y=which[i];
                    if (i%sq!=1)
                        {
                            Solve(i,min(i+sq-1,stop[x]),which[i],s);
                            i=which[i]*sq+1;
                        }
                    while ((i+sq-1)<=stop[x]) {add[which[i]]+=s;i+=sq;}
                    if (i<=stop[x]) Solve(i,stop[x],which[i],s);
                }
            if (op==2)
                {
                    fin>>s;
                    aux=0;poz=-1;
                    for (i=1;i<=n;i+=sq)
                        {
                            aux++;
                            if (M[aux].find(s-add[aux])!=M[aux].end())
                                poz=M[aux][s-add[aux]];
                        }
                    fout<<poz<<"\n";
                }
        }
    return 0;
}