Cod sursa(job #1451125)

Utilizator ZenusTudor Costin Razvan Zenus Data 16 iunie 2015 10:52:09
Problema Arbore Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.63 kb
#include <bits/stdc++.h>

using namespace std;

#define wN second.first
#define wP second.second
#define wI first

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

const int SSIZE = 317 + 10;
const int NSIZE = 100000 + 10;

pair < int , int > p[NSIZE];
int add[SSIZE] , aux[NSIZE];
vector < pair < int , pair < int , int > > > iB[SSIZE];
int N , Q , x , y , type , i , crt , block;
vector < int > g[NSIZE];

int wB(int i)
{
    return i / block;
}

void df(int node,int fa)
{
    p[node].first = ++crt;
    aux[crt] = node;

    for (int i=0;i<g[node].size();++i)
    {
        int to = g[node][i];

        if (to != fa) df(to,node);
    }

    p[node].second = crt;
}

void build()
{
    int i;
    for (i = 0 ; i < N ; ++i)
    iB[wB(i)].push_back({0 , {aux[i] , i}});

    for (i = 0 ; i <= wB(N - 1) ; ++i)
    sort(iB[i].begin() , iB[i].end());
}

void update(int le,int ri,int x)
{
    int i , j1 , j2 , crt;

    if (le % block != 0)
    {
        j1 = wB(le) + 1;
        crt = wB(le);
        i = le;
        while (i <= ri && i % block != 0)
        {
            if (le <= iB[crt][i].wP && iB[crt][i].wP <= ri)
            iB[crt][i].wI += x;

            i += 1;
        }

        sort(iB[crt].begin(),iB[crt].end());

        if (wB(le) == wB(ri)) return ;
    } else j1 = wB(le);

    if (ri % block != 1)
    {
        j2 = wB(ri) - 1;
        crt = wB(ri);
        i = ri;
        while (i >= le && i % block != 1)
        {
            if (le <= iB[crt][i].wP && iB[crt][i].wP <= ri)
            iB[crt][i].wI += x;

            i -= 1;
        }

        sort(iB[crt].begin(),iB[crt].end());

        if (wB(le) == wB(ri)) return ;
    } else j2 = wB(ri);

    for (i = j1 ; i <= j2 ; ++i)
    add[i] += x;
}

int query(int x)
{
    int i , j1 , j2 , k;

    for (i = 0 ; i <= wB(N - 1) ; ++i)
    {
        j1 = 0 , j2 = iB[i].size() - 1;

        while (j1 <= j2)
        {
            k = (j1 + j2) >> 1;

            if (iB[i][k].wI + add[i] == x)
            return iB[i][k].wN;

            if (iB[i][k].wI + add[i] < x)
            j1 = k + 1;
            else j2 = k - 1;
        }
    }

    return -1;
}

int main()
{

fin >> N >> Q;
block = (int)sqrt(1.0 * N);

for (i=1;i<N;++i)
{
    fin >> x >> y;
    g[x].push_back(y);
    g[y].push_back(x);
}

crt = -1;
df(1,0);
build();

for (i=1;i<=Q;++i)
{
    fin >> type;

    if (type == 1)
    {
        fin >> x >> y;
        update(p[x].first,p[x].second,y);
    }
    else
    {
        fin >> y;
        fout << query(y) << '\n';
    }
}

return 0;
}