제출 #1334123

#제출 시각아이디문제언어결과실행 시간메모리
1334123kawhiet말 (IOI15_horses)C++20
0 / 100
268 ms63052 KiB
#include <bits/stdc++.h>
#include "horses.h"
using namespace std;

using i64 = long long;

constexpr int mod = 1e9 + 7;

struct SegmentTree {
    int n;
    vector<i64> t;

    void init(int _n) {
        n = _n;
        t.assign(4 * n, 1);
    }

    void update(int id, int tl, int tr, int i, i64 v) {
        if (tl == tr) {
            t[id] = v;
            return;
        }
        int x = (id << 1) + 1, y = x + 1, tm = (tl + tr) >> 1;
        if (i <= tm) {
            update(x, tl, tm, i, v);
        } else {
            update(y, tm + 1, tr, i, v);
        }
        t[id] = t[x] * t[y] % mod;
    }

    i64 get(int id, int tl, int tr, int l, int r) {
        if (r < tl || tr < l) return 1;
        if (l <= tl && tr <= r) return t[id];
        int x = (id << 1) + 1, y = x + 1, tm = (tl + tr) >> 1;
        return get(x, tl, tm, l, r) * get(y, tm + 1, tr, l, r) % mod;
    }

    void update(int i, int v) { update(0, 0, n - 1, i, v); }
    i64 get(int l, int r) { return get(0, 0, n - 1, l, r); }
};

struct mxSegmentTree {
    int n;
    vector<i64> t;

    void init(int _n) {
        n = _n;
        t.resize(4 * n);
    }

    int merge(int x, int y) {
        return max(x, y);
    }

    void update(int id, int tl, int tr, int i, int v) {
        if (tl == tr) {
            t[id] = v;
            return;
        }
        int x = (id << 1) + 1, y = x + 1, tm = (tl + tr) >> 1;
        if (i <= tm) {
            update(x, tl, tm, i, v);
        } else {
            update(y, tm + 1, tr, i, v);
        }
        t[id] = merge(t[x], t[y]);
    }

    i64 get(int id, int tl, int tr, int l, int r) {
        if (r < tl || tr < l) return 0;
        if (l <= tl && tr <= r) return t[id];
        int x = (id << 1) + 1, y = x + 1, tm = (tl + tr) >> 1;
        return merge(get(x, tl, tm, l, r), get(y, tm + 1, tr, l, r));
    }

    void update(int i, int v) { update(0, 0, n - 1, i, v); }
    int get(int l, int r) { return get(0, 0, n - 1, l, r); }
};

int n;
SegmentTree t;
mxSegmentTree mx;
vector<int> x, y;

set<int> s;

// is r better than l

bool is(int l, int r) {
    i64 g = mx.get(l, n - 1);
    i64 k = mx.get(r, n - 1);
    for (int i = l + 1; i <= r; i++) {
        k *= x[i];
        if (k > g) {
            return true;
        }
    }
    return false;
}

int solve() {
    auto it = s.end();
    vector<array<int, 2>> b;
    for (int _ = 0; _ < 40; _++) {
        if (it == s.begin()) break;
        --it;
        int j = *it;
        b.push_back({j, mx.get(j, n - 1)});
    }
    if (b.empty()) {
        return mx.get(0, n - 1);
    }
    ranges::reverse(b);
    int pos = b[0][0];
    i64 ans = t.get(0, b[0][0]) * b[0][1] % mod;
    for (int i = 1; i < n; i++) {
        if (is(pos, i)) {
            pos = i;
            ans = t.get(0, b[i][0]) * b[i][1] % mod;
        }
    }
    return ans;
}

int init(int N, int X[], int Y[]) {
    n = N;
    x.assign(X, X + N);
    y.assign(Y, Y + N);
    t.init(n);
    mx.init(n);
    for (int i = 0; i < n; i++) {
        if (x[i] > 1) {
            s.insert(i);
        }
        mx.update(i, y[i]);
        t.update(i, x[i]);
    }
    return solve();
}

int updateX(int pos, int val) {
    if (x[pos] > 1) {
        s.erase(pos);
    }
    if (val > 1) {
        s.insert(pos);
    }
    t.update(pos, val);
    x[pos] = val;
    return solve();
}

int updateY(int pos, int val) {
    y[pos] = val;
    mx.update(pos, val);
    return solve();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...