제출 #308560

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
308560		aZvezda	Birthday gift (IZhO18_treearray)	C++14	56 / 100	1047 ms	262144 KiB

treearray

#include <bits/stdc++.h>
using namespace std;
//#pragma GCC optimize ("O3")
//#pragma GCC target ("sse4")
#define endl "\n"
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
template<class T, class T2> inline bool chkmax(T &x, const T2 &y) { return x < y ? x = y, 1 : 0; }
template<class T, class T2> inline bool chkmin(T &x, const T2 &y) { return x > y ? x = y, 1 : 0; }
const ll mod = 1e9 + 7;
template<class T> inline void fix(T &x) {if(labs(x) >= mod) {x %= mod;} if(x < 0) {x += mod;}}
#define out(x) cout << __LINE__ << ": " << (#x) << " = " << (x) << endl

const int MAX_N = 2e5 + 10;

struct MergeSortTree {
    set<pair<int, int> > tree[MAX_N];
    int query(int curr, int l, int r, int ql, int qr, int v) {
        if(ql <= l && r <= qr) {
            auto it = tree[curr].lower_bound({v, -1});
            if(it == tree[curr].end() || (*it).first != v) {
                return -1;
            } else {
                return (*it).second;
            }
        } else if(l > qr || r < ql) {return -1;}
        int m = (l + r) / 2ll;
        int currAns = query(curr * 2, l, m, ql, qr, v);
        if(currAns != -1) {return currAns;}
        currAns = query(curr * 2 + 1, m + 1, r, ql, qr, v);
        if(currAns != -1) {return currAns;}
        return -1;
    }
    void change(int curr, int l, int r, int ind, int prv, int now) {
        if(l > ind || ind > r) {return;}
        auto it = tree[curr].find(pair<int, int>(prv, ind)); if(it != tree[curr].end()) {tree[curr].erase(it);}
        tree[curr].insert(pair<int, int>(now, ind));
        if(l == r) {return;}
        int m = (l + r) / 2ll;
        change(curr * 2, l, m, ind, prv, now);
        change(curr * 2 + 1, m + 1, r, ind, prv, now);
    }
    void print(int curr, int l, int r) {
        return;
        cout << curr << " " << l << " " << r << ": ";
        for(auto it : tree[curr]) {cout << "{" << it.first << ", " << it.second << "} ";}
        cout << endl;
        if(l == r) {return;}
        int m = (l + r) / 2ll;
        print(curr * 2, l, m);
        print(curr * 2 + 1, m + 1, r);
    }
};

MergeSortTree diff, seq;

const int LOG = 20;
int par[MAX_N][LOG], d[MAX_N];
vector<int> g[MAX_N];

void dfs(int x, int p) {
    par[x][0] = p;
    d[x] = d[p] + 1;
    for(int i = 1; i < LOG; i ++) {
        par[x][i] = par[par[x][i - 1]][i - 1];
    }
    for(auto it : g[x]) {
        if(it == p) {continue;}
        dfs(it, x);
    }
}

int lca(int a, int b) {
    if(d[a] < d[b]) {swap(a, b);}
    for(int i = LOG - 1; i >= 0; i --) {
        if(d[par[a][i]] >= d[b]) {
            a = par[a][i];
        }
    }
    if(a == b) {return a;}
    for(int i = LOG - 1; i >= 0; i --) {
        if(par[a][i] != par[b][i]) {
            a = par[a][i];
            b = par[b][i];
        }
    }
    return par[a][0];
}

int n, m, q;
int arr[MAX_N];

signed main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    cin >> n >> m >> q;
    for(int i = 0; i < n - 1; i ++) {
        int a, b;
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    dfs(1, 0);
    for(int i = 1; i <= m; i ++) {
        cin >> arr[i];
        seq.change(1, 1, m, i, 0, arr[i]);
    }
    for(int i = 1; i <= m - 1; i ++) {
        int curr = lca(arr[i], arr[i + 1]);
        diff.change(1, 1, m, i, 0, curr);
    }
    while(q --) {
        int t;
        cin >> t;
        if(t == 1) {
            int a, v;
            cin >> a >> v;
            seq.change(1, 1, m, a, arr[a], v);
            diff.print(1, 1, m);
            if(a != m) {
                diff.change(1, 1, m, a, lca(arr[a], arr[a + 1]), lca(v, arr[a + 1]));
            }
            diff.print(1, 1, m);
            if(a != 1) {
                diff.change(1, 1, m, a - 1, lca(arr[a], arr[a - 1]), lca(v, arr[a - 1]));
            }
            arr[a] = v;
            diff.print(1, 1, m);
        } else {
            int l, r, v;
            cin >> l >> r >> v;
            int ans = seq.query(1, 1, m, l, r, v);
            if(ans != -1) {
                cout << ans << " " << ans << endl;
                continue;
            }
            ans = diff.query(1, 1, m, l, r - 1, v);
            if(ans != -1) {
                cout << ans << " " << ans + 1 << endl;
                continue;
            }
            cout << -1 << " " << -1 << endl;
        }
    }
    return 0;
}

/*
5 4 4
1 2
3 1
3 4
5 3
4 5 2 3
2 1 3 1
1 3 5
2 3 4 5
2 1 3 1
*/

• Subtask 1: 12 / 12
• Subtask 2: 18 / 18
• Subtask 3: 26 / 26
• Subtask 4: 0 / 44