Cod sursa(job #2384466)

Utilizator flaviu_2001Craciun Ioan-Flaviu flaviu_2001 Data 20 martie 2019 19:33:03
Problema Arbore Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 4.19 kb
#include <bits/stdc++.h>
#define ff first
#define ss second

using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pi;

const ll INF = 9223372036854775807ll;
const int inf = 2147483647, B = 400, nmax = 100005, VMAX = 1000006;

class InParser {
private:
    FILE *fin;
    char *buff;
    int sp;

    char read_ch() {
        ++sp;
        if (sp == 4096) {
            sp = 0;
            fread(buff, 1, 4096, fin);
        }
        return buff[sp];
    }

public:
    InParser(const char* nume) {
        fin = fopen(nume, "r");
        buff = new char[4096]();
        sp = 4095;
    }

    InParser& operator >> (int &n) {
        char c;
        while (!isdigit(c = read_ch()) && c != '-');
        int sgn = 1;
        if (c == '-') {
            n = 0;
            sgn = -1;
        } else {
            n = c - '0';
        }
        while (isdigit(c = read_ch())) {
            n = 10 * n + c - '0';
        }
        n *= sgn;
        return *this;
    }
};

class OutParser {
private:
    FILE *fout;
    char *buff;
    int sp;

    void write_ch(char ch) {
        if (sp == 50000) {
            fwrite(buff, 1, 50000, fout);
            sp = 0;
            buff[sp++] = ch;
        } else {
            buff[sp++] = ch;
        }
    }


public:
    OutParser(const char* name) {
        fout = fopen(name, "w");
        buff = new char[50000]();
        sp = 0;
    }
    ~OutParser() {
        fwrite(buff, 1, sp, fout);
        fclose(fout);
    }

    OutParser& operator << (int vu32) {
        if(vu32 == -1){
            write_ch('-');
            write_ch('1');
        } else if (vu32 <= 9) {
            write_ch(vu32 + '0');
        } else {
            (*this) << (vu32 / 10);
            write_ch(vu32 % 10 + '0');
        }
        return *this;
    }

    OutParser& operator << (char ch) {
        write_ch(ch);
        return *this;
    }
    OutParser& operator << (const char *ch) {
        while (*ch) {
            write_ch(*ch);
            ++ch;
        }
        return *this;
    }
};


int n, test, e[nmax], first[nmax], last[nmax], p, add[nmax], v[nmax];
vector<int> g[nmax];
bitset<VMAX> ok[nmax/B+5];

void dfs(int x, int pred = 0)
{
    e[++p] = x;
    first[x] = p;
    for (auto y : g[x])
        if(y != pred)
            dfs(y, x);
    last[x] = p;
}

void resetBucket(int a)
{
    for (int i = max((a/B)*B, 1); i/B == a/B && i <= n; ++i)
        ok[a/B][v[i]] = 0;
    for (int i = max((a/B)*B, 1); i/B == a/B && i <= n; ++i){
        v[i] += add[a/B];
        ok[a/B][v[i]] = 1;
    }
    add[a/B] = 0;
}

int main()
{
    InParser fin ("arbore.in");
    OutParser fout ("arbore.out");
    fin >> n >> test;
    for (int i = 1; i < n; ++i){
        int x, y;
        fin >> x >> y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1);
    while(test--){
        int t;
        fin >> t;
        if(t == 1){
            int nod, val;
            fin >> nod >> val;
            int a = first[nod], b = last[nod];
            if(a/B == b/B){
                for (int i = a; i <= b; ++i){
                    ok[a/B][v[i]] = 0;
                    v[i] += val;
                }
                resetBucket(a);
            }else{
                for (int i = a; i/B == a/B && i <= n; ++i){
                    ok[a/B][v[i]] = 0;
                    v[i] += val;
                }
                resetBucket(a);
                a = ((a+B)/B)*B;
                for (int i = b; i/B == b/B && i >= 1; --i){
                    ok[b/B][v[i]] = 0;
                    v[i] += val;
                }
                resetBucket(b);
                b = (b/B)*B;
                while(a != b){
                    add[a/B] += val;
                    a += B;
                }
            }
        }else{
            int val, ans = -1;
            fin >> val;
            for (int b = 0; b*B <= n; ++b)
                if(val-add[b] >= 0 && ok[b][val-add[b]]){
                    for (int i = max(b*B, 1); i/B == b && i <= n; ++i)
                        if(v[i]+add[b] == val){
                            ans = e[i];
                            break;
                        }
                    break;
                }
            fout << ans << "\n";
        }
    }
    return 0;
}