AcWing 1275. 最大数

\(AcWing\) \(1275\). 最大数

一、题目描述

给定一个正整数数列 \(a_1,a_2,…,a_n\),每一个数都在 \(0∼p−1\) 之间。

可以对这列数进行两种操作:

添加操作:向序列后添加一个数,序列长度变成 \(n+1\)
询问操作:询问这个序列中最后 \(L\) 个数中最大的数是多少。

程序运行的最开始,整数序列为空。

一共要对整数序列进行 \(m\) 次操作。

写一个程序,读入操作的序列,并输出询问操作的答案。

输入格式
第一行有两个正整数 \(m,p\),意义如题目描述;

接下来 \(m\) 行,每一行表示一个操作。

如果该行的内容是 \(Q\) \(L\),则表示这个操作是询问序列中最后 \(L\) 个数的最大数是多少;

如果是 \(A\) \(t\),则表示向序列后面加一个数,加入的数是 \((t+a)\) \(mod\) \(p\)。其中,\(t\) 是输入的参数,\(a\) 是在这个添加操作之前最后一个询问操作的答案(如果之前没有询问操作,则 \(a=0\))。

第一个操作一定是添加操作。对于询问操作,\(L>0\) 且不超过当前序列的长度。

输出格式
对于每一个询问操作,输出一行。该行只有一个数,即序列中最后 \(L\) 个数的最大数。

数据范围
\(1≤m≤2×10^5,1≤p≤2×10^9,0≤t<p\)

输入样例

10 100
A 97
Q 1
Q 1
A 17
Q 2
A 63
Q 1
Q 1
Q 3
A 99

输出样例

97
97
97
60
60
97

样例解释
最后的序列是 \(97,14,60,96\)

二、树状数组解法

运行时间: \(246\) \(ms\)

重点: 因为每次询问的区间是最后几个数字,而树状数组可以快速求前缀,所以不妨倒着存储数据

为了减少时间复杂度,要提前预留\(m\)个坑位,第一个数存放到\(m\),第二个数存放到\(m-1\),以此类推

#include <iostream>
#include <cstdio>
typedef long long LL;
using namespace std;
const int N = 2e5 + 10;
int tr[N];

int m, p;

#define lowbit(x) (x & -x)
void add(int n, int x) {
    for (int i = n; i <= m; i += lowbit(i)) tr[i] = max(tr[i], x);
}
int query(int n) {
    int ret = 0;
    for (int i = n; i; i -= lowbit(i)) ret = max(ret, tr[i]);
    return ret;
}

int main() {
    //加快读入
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> m >> p;
    int last = 0;
    int j = m;
    for (int i = 0; i < m; i++) {
        LL x;
        char c;
        cin >> c >> x;
        if (c == 'A')
            add(j--, (x + last) % p);
        else {
            last = query(j + x);
            printf("%d\n", last);
        }
    }
    return 0;
}

三、线段树求最大值模板代码

运行时间:\(646\) \(ms\)

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int N = 200010;
int m; // m个操作
int p; // mod p

// 线段树求最大值模板
#define ls (u << 1)
#define rs (u << 1 | 1)
#define mid ((l + r) >> 1)
struct Node {
    int l, r, len;
    int v;
} tr[N << 2];

void pushup(int u) {
    tr[u].v = max(tr[ls].v, tr[rs].v);
}

void build(int u, int l, int r) {
    tr[u].l = l, tr[u].r = r;
    if (l == r) return;
    build(ls, l, mid), build(rs, mid + 1, r);
}

int query(int u, int L, int R) {
    int l = tr[u].l, r = tr[u].r;
    if (l >= L && r <= R) return tr[u].v;
    if (l > R || r < L) return 0;
    return max(query(ls, L, R), query(rs, L, R));
}

void modify(int u, int x, int v) {
    int l = tr[u].l, r = tr[u].r;
    if (l == r) {
        tr[u].v = v;
        return;
    }
    if (x <= mid)
        modify(u << 1, x, v);
    else
        modify(u << 1 | 1, x, v);
    pushup(u);
}

int n, last;

int main() {
    // 加快读入
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> m >> p;

    // 单点修改,求区间最大值

    // ① 初始化线段树,最多m次操作,最多m个数,区间范围[1, m]
    build(1, 1, m);

    int x;
    char op;
    while (m--) {
        cin >> op >> x;
        if (op == 'A') {
            // 对于节点++n进行修改,值=(最后一次的last查询值 + x )%p
            modify(1, ++n, ((LL)last + x) % p);
        } else {
            // u = 1:从根节点开始查询
            // 查询序列中最后x个数的最大数: 查询[n - x + 1, n]内的最大值
            last = query(1, n - x + 1, n);
            printf("%d\n", last);
        }
    }
    return 0;
}

四、\(ST\)表解法

大部分大佬用的是线段树(我第一次也用的线段树)
现在发现用\(st\)超级容易
每次插入只需要修改与最后一个点有关的值即可
典型的空间换时间;
线段树空间复杂度\(O(4N)\),查询插入都是\(O(logN)\),不过时间里面还有许多浪费的成分
\(st\)空间复杂度\(O(NlogN)\),查询\(O(1)\),插入\(O(logN)\),比线段树快至少一半

由于\(st\)表还没有复习到,这里先不深入研究\(st\)表的解法,后面第三刷时再来仔细研究:
\(2022-12-13\) 新冠非常猖獗的时候,记录一下生活~

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=2e5+10,M=log2(N)+10;
int f[N][M];
int query(int l,int r){
    int k=log2(r-l+1);
    return max(f[l][k],f[r-(1<<k)+1][k]);
}
int n;
int insert(int a){
    f[++n][0]=a;
    for(int i=1;1<<i<=n;i++)
        f[n-(1<<i)+1][i]=max(f[n-(1<<i)+1][i-1],f[n-(1<<i-1)+1][i-1]);
}
int q,p,a;
char op[2];
int x;
int main(){
    scanf("%d%d",&q,&p);
    for(;q--;){
        scanf("%s%d",op,&x);
        if(op[0]=='A')
            insert((ll(x)+a)%p);
        else{
            printf("%d\n",a=query(n-x+1,n));
        }
    }
    return 0;
}

posted @ 2022-04-16 11:18  糖豆爸爸  阅读(178)  评论(2编辑  收藏  举报
Live2D