acwing-243. 一个简单的整数问题2(线段树+双懒惰标记)

最新推荐文章于 2024-03-29 09:15:13 发布

cc_奋斗ing

最新推荐文章于 2024-03-29 09:15:13 发布

阅读量91

点赞数

分类专栏： ACM 文章标签： acm竞赛

本文链接：https://blog.csdn.net/m0_46656833/article/details/116489724

版权

ACM 专栏收录该内容

147 篇文章 2 订阅

订阅专栏

本文介绍了一个在线算法实现，用于处理数列的区间加法操作和区间查询求和问题。利用树状数组（也称为线段树）的数据结构，能够高效地维护数列并回答各种查询。代码示例展示了如何构建、更新和查询线段树，以处理大量输入和查询，确保在限定的时间和空间复杂度内完成任务。

摘要由CSDN通过智能技术生成

给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
Q l r，表示询问数列中第 l∼r 个数的和。
对于每个询问，输出一个整数表示答案。

输入格式
第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式
对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围
1≤N,M≤105,
|d|≤10000,
|A[i]|≤109

输入样例：
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
输出样例：
4
55
9
15

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lson u << 1
#define rson u << 1 | 1
const int N = 1e5 + 10;
struct {
    int l,r;
    int sum;
    int mul,add;
}tr[4 * N];
int a[N];
int n,p,m;
void eval(int u,int add,int mul){
    tr[u].sum = (((ll)mul * tr[u].sum) % p + ((ll)add * (tr[u].r - tr[u].l + 1) % p)) % p;
    tr[u].add = (((ll)tr[u].add * mul) % p + add) % p;
    tr[u].mul = ((ll)tr[u].mul * mul) % p;
}
void pushdown(int u){
    eval(lson,tr[u].add,tr[u].mul);
    eval(rson,tr[u].add,tr[u].mul);
    tr[u].mul = 1,tr[u].add = 0;
}
void pushup(int u){
    tr[u].sum = (tr[lson].sum + tr[rson].sum) % p;
}
void build(int u,int l,int r){
    if(l == r)tr[u] = {l,r,a[l],1,0};
    else{
        tr[u] = {l,r,0,1,0};
        int mid = (l + r) >> 1;
        build(u << 1,l,mid);
        build(u << 1 | 1,mid + 1,r);
        pushup(u);
    }
}
int query(int u,int l,int r){
    if(tr[u].l >= l && tr[u].r <= r)return tr[u].sum;
    else{
        pushdown(u);
        int mid = (tr[u].l + tr[u].r) >> 1;
        if(r <= mid)return query(lson,l,r);
        else if(l > mid)return query(rson,l,r);
        else return (query(lson,l,r) + query(rson,l,r)) % p;
    }
}
void modify(int u,int l,int r,int add,int mul){
    if(tr[u].l >= l && tr[u].r <= r)eval(u,add,mul);
    else{
        pushdown(u);
        int mid = (tr[u].l + tr[u].r) >> 1;
        if(l <= mid)modify(lson,l,r,add,mul);
        if(r > mid)modify(rson,l,r,add,mul);
        pushup(u);
    }
}
int main(){
    scanf("%d %d",&n,&p);
    for(int i = 1;i <= n;i ++)scanf("%d",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    char c;
    int l,r,a;
    for(int i = 0;i < m;i ++){
        scanf(" %c %d %d",&c,&l,&r);
        if(c == '1'){
            scanf("%d",&a);
            modify(1,l,r,0,a);
        }
        else if(c == '2'){
            scanf("%d",&a);
            modify(1,l,r,a,1);
        }else printf("%d\n",query(1,l,r));
    }
    return 0;
}