【线段树】ACW1275. 最大数

题目

解析

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>
#include <functional>
#include <unordered_set>
#include <unordered_map>
#include <set>
#include <bitset>


using namespace std;

const int N = 2e5 + 100;

int m, p;

struct Node { // 线段树节点
    int l, r; // 左右端点
    int v; // 区间维护的值
} tr[N * 4]; // 通常开四倍区间

inline void pushup(int u) {
    // 用子节点向上传递更新信息
    tr[u].v = max(tr[u << 1].v, tr[u << 1 | 1].v);
}

void build(int u, int l, int r) {
    tr[u] = {l, r};
    if (l == r) return;
    int mid = l + r >> 1;
    build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
}

int query(int u, int l, int r) {
    // 查询u在[l,r]区间的值
    // u在[l,r]区间内则直接返回信息
    if (tr[u].l >= l && tr[u].r <= r) return tr[u].v;

    // 否则递归查询
    int mid = tr[u].l + tr[u].r >> 1;
    int v = 0;
    if (l <= mid) v = query(u << 1, l, r); // 与左区间有接触，在做区间查询
    if (r > mid) v = max(v, query(u << 1 | 1, l, r)); // 与右区间有接触，在右侧接着查询
    return v;
}

void modify(int u, int x, int v) {
    if (tr[u].l == x && tr[u].r == x) tr[u].v = v;
    else {
        int mid = tr[u].l + tr[u].r >> 1;
        if (mid >= x) modify(u << 1, x, v); // x在左侧区间
        else modify(u << 1 | 1, x, v); // x在右侧区间
        // 递归更新
        pushup(u);
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin >> m >> p;
    build(1, 1, m);
    string s;
    int num;
    int last = 0, n = 0;
    for (int i = 1; i <= m; i++) {
        cin >> s >> num;
        if (s[0] == 'Q') {
            last = query(1, n - num + 1, n);
            cout << last << endl;
        } else {
            modify(1, n + 1, (num + last) % p);
            n++;
        }
    }
    return 0;
}