Cod sursa(job #2224654)

Utilizator Andrei_CotorAndrei Cotor Andrei_Cotor Data 24 iulie 2018 19:55:27
Problema Arbore Scor 45
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.73 kb
#include<fstream>
#include<vector>
#include<bitset>
#define Sz 317
using namespace std;
ifstream fi("arbore.in");
ofstream fo("arbore.out");
vector<int> G[100005];
int n,m,i,tip,x,y,nrg,St[100005],Dr[100005],A[100005],ind,Val[100005],Sum[Sz+5];
bitset<100005> B[Sz+5];

void dfs(int nod, int p)
{
    vector<int>::iterator it;
    A[++ind]=nod;
    St[nod]=ind;
    for(it=G[nod].begin(); it!=G[nod].end(); it++)
        if((*it)!=p)
            dfs(*it,nod);
    Dr[nod]=ind;
}

void upd(int st, int dr, int val)
{
    int i;
    int g=st/Sz;
    for(i=st; i<min(Sz*(g+1),dr+1); i++)
    {
        B[g][Val[i]]=0;
        Val[i]=Val[i]+val;
    }
    for(i=Sz*g; i<(g+1)*Sz; i++)
        B[g][Val[i]]=1;
    g++;
    while(g<(dr/Sz))
    {
        Sum[g]+=val;
        g++;
    }
    for(i=g*Sz; i<=dr; i++)
    {
        B[g][Val[i]]=0;
        Val[i]=Val[i]+val;
    }
    for(i=Sz*g; i<(g+1)*Sz; i++)
        B[g][Val[i]]=1;
}

int qry(int s)
{
    int i,j;
    for(i=0; i<nrg; i++)
        if(Sum[i]<=s && B[i][s-Sum[i]])
            for(j=i*Sz; j<(i+1)*Sz; j++)
                if(Val[j]+Sum[i]==s)
                    return A[j];
    for(i=Sz*nrg; i<=n; i++)
        if(Val[i]+Sum[nrg]==s)
            return A[i];
    return -1;
}

int main()
{
    fi>>n>>m;
    for(i=1; i<n; i++)
    {
        fi>>x>>y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(1,0);
    nrg=n/Sz;
    for(i=0; i<=nrg; i++)
        B[i][0]=1;
    for(i=1; i<=m; i++)
    {
        fi>>tip>>x;
        if(tip==1)
        {
            fi>>y;
            upd(St[x],Dr[x],y);
        }
        else
            fo<<qry(x)<<"\n";
    }
    fi.close();
    fo.close();
    return 0;
}