ACM模板线段树

Posted 2021-05-23 nefu_ljw

P3372 【模板】线段树 1

区间修改（加法），区间查询（和）。

#include <bits/stdc++.h>
#define lson i<<1
#define rson i<<1|1
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll a[N];
struct node
{
    int l,r;
    ll lazy; // 懒标记(存每个结点需要加上的和，如果为0则为无标记)
    ll val; // 和
}tree[N<<2];
void pushup(int i)
{
    tree[i].val=tree[lson].val+tree[rson].val;
}
void build(int i,int l,int r)
{
    tree[i].l=l;
    tree[i].r=r;
    tree[i].lazy=0;
    if(l==r)
    {
        tree[i].val=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
    pushup(i);
}
void add(int i,ll k) // 更新标记和结点i的val(和)
{
    tree[i].lazy+=k;
    tree[i].val+=(tree[i].r-tree[i].l+1)*k;
}
void pushdown(int i)
{
    if(tree[i].lazy)
    {
        // 标记下传、更新左右孩子的和
        add(lson,tree[i].lazy);
        add(rson,tree[i].lazy);
        // 清除标记
        tree[i].lazy=0;
    }
}
void update(int i,int l,int r,ll k)
{
    if(tree[i].r<l||tree[i].l>r)return;
    if(tree[i].l>=l&&tree[i].r<=r)return add(i,k); // 更新自己的标记和val(和)
    pushdown(i);
    update(lson,l,r,k);
    update(rson,l,r,k);
    pushup(i);
}
ll query(int i,int l,int r)
{
    if(tree[i].r<l||tree[i].l>r)return 0;
    if(tree[i].l>=l&&tree[i].r<=r)return tree[i].val;
    pushdown(i);
    return query(lson,l,r)+query(rson,l,r);
}
int main()
{
    ios::sync_with_stdio(false);
    int n,m,opt,l,r;
    ll k;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    build(1,1,n);
    while(m--)
    {
        cin>>opt;
        if(opt==1)
        {
            cin>>l>>r>>k;
            update(1,l,r,k);
        }
        else
        {
            cin>>l>>r;
            printf("%lld\\n",query(1,l,r));
        }
    }
    return 0;
}

P3373 【模板】线段树 2

区间修改（乘法、加法），区间查询（和）。

注意：

• 乘法标记初值为1。
• 区间乘法操作时要修改 乘法标记、区间和、加法标记！
• 区间加法操作时只修改 加法标记、区间和。

#include <bits/stdc++.h>
#define lson i<<1
#define rson i<<1|1
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll mod,a[N];
struct node
{
    int l,r;
    ll lazy_add,lazy_mul,val;
}tree[N<<2];
void pushup(int i)
{
    tree[i].val=tree[lson].val+tree[rson].val;
}
void build(int i,int l,int r)
{
    tree[i].l=l;
    tree[i].r=r;
    tree[i].lazy_add=0;
    tree[i].lazy_mul=1;  // 注意：乘法标记初始值为1
    if(l==r)
    {
        tree[i].val=a[l]%mod;
        return;
    }
    int mid=(l+r)>>1;
    build(lson,l,mid);
    build(rson,mid+1,r);
    pushup(i);
}
void add(int i,ll k)
{
    tree[i].lazy_add=(tree[i].lazy_add+k)%mod;
    tree[i].val=(tree[i].val+(tree[i].r-tree[i].l+1)*k%mod)%mod;
}
void mul(int i,ll k)
{
    tree[i].lazy_mul=tree[i].lazy_mul*k%mod;
    tree[i].lazy_add=tree[i].lazy_add*k%mod; //注意：加法标记也要乘k
    tree[i].val=tree[i].val*k%mod;
}
void pushdown(int i)
{
    if(tree[i].lazy_mul!=1)  // 注意：这里是 !=1 而不是 !=0 
    {
        mul(lson,tree[i].lazy_mul);
        mul(rson,tree[i].lazy_mul);
        tree[i].lazy_mul=1;
    }
    if(tree[i].lazy_add!=0)
    {
        add(lson,tree[i].lazy_add);
        add(rson,tree[i].lazy_add);
        tree[i].lazy_add=0;
    }
}
void update(int i,int l,int r,ll k,int opt)
{
    if(tree[i].r<l||tree[i].l>r)return;
    if(tree[i].l>=l&&tree[i].r<=r)
    {
        if(opt==1)return mul(i,k);
        else if(opt==2)return add(i,k);
    }
    pushdown(i);
    update(lson,l,r,k,opt);
    update(rson,l,r,k,opt);
    pushup(i);
}
ll query(int i,int l,int r)
{
    if(tree[i].r<l||tree[i].l>r)return 0;
    if(tree[i].l>=l&&tree[i].r<=r)return tree[i].val;
    pushdown(i);
    return (query(lson,l,r)+query(rson,l,r))%mod;
}
int main()
{
    ios::sync_with_stdio(false);
    int n,m,opt,l,r;
    ll k;
    cin>>n>>m>>mod;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    build(1,1,n);
    while(m--)
    {
        cin>>opt;
        if(opt==3)
        {
            cin>>l>>r;
            printf("%lld\\n",query(1,l,r));
        }
        else
        {
            cin>>l>>r>>k;
            update(1,l,r,k,opt);
        }
    }
    return 0;
}