P3373 线段树模板

好，这是一个线段树模板。

#include <cstdio>
using namespace std;
const long long int N=1000010;
long long int sum[N],tag1[N],tag2[N],mo;

void build(int l,int r,int o)
{
    tag1[o]=1;
    if(l==r)
    {
        scanf ("%d",&sum[o]);
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,o<<1);
    build(mid+1,r,o<<1|1);
    sum[o]=(sum[o<<1]+sum[o<<1|1])%mo;
    return;
}
void down(int l,int r,int o)
{
    tag1[o<<1]=tag1[o<<1]*tag1[o]%mo;
    tag1[o<<1|1]=tag1[o<<1|1]*tag1[o]%mo;
    tag2[o<<1]=(tag2[o<<1]*tag1[o]+tag2[o])%mo;
    tag2[o<<1|1]=(tag2[o<<1|1]*tag1[o]+tag2[o])%mo;
    int mid=(l+r)>>1;
    sum[o<<1]=(sum[o<<1]*tag1[o]+tag2[o]*(mid-l+1))%mo;
    sum[o<<1|1]=(sum[o<<1|1]*tag1[o]+tag2[o]*(r-mid))%mo;
    tag1[o]=1;
    tag2[o]=0;
    return;
}

void add(int L,int R,int v,int l,int r,int o)
{
    if(L<=l && r<=R)
    {
        tag2[o]=(tag2[o]+v)%mo;
        sum[o]=(sum[o]+v*(r-l+1))%mo;
        return;
    }
    if(r<L || R<l) return;
    down(l,r,o);
    int mid=(l+r)>>1;
    if(L<=mid) add(L,R,v,l,mid,o<<1);
    if(mid<R)  add(L,R,v,mid+1,r,o<<1|1);
    sum[o]=(sum[o<<1]+sum[o<<1|1])%mo;
    return;
}
void mul(int L,int R,int v,int l,int r,int o)
{
    if(L<=l && r<=R)
    {
        tag1[o]=tag1[o]*v%mo;
        tag2[o]=tag2[o]*v%mo;
        sum[o]=sum[o]*v%mo;
        return;
    }
    if(r<L || R<l) return;
    down(l,r,o);
    int mid=(l+r)>>1;
    if(L<=mid) mul(L,R,v,l,mid,o<<1);
    if(mid<R)  mul(L,R,v,mid+1,r,o<<1|1);
    sum[o]=(sum[o<<1]+sum[o<<1|1])%mo;
    return;
}
long long int ask_sum(int L,int R,int l,int r,int o)
{
    if(L<=l && r<=R) return sum[o];
    if(r<L || R<l)   return 0;
    int mid=(l+r)>>1;
    down(l,r,o);
    long long int ans=ask_sum(L,R,l,mid,o<<1);
    ans=ans+ask_sum(L,R,mid+1,r,o<<1|1)%mo;
    sum[o]=(sum[o<<1]+sum[o<<1|1])%mo;
    return ans%mo;
}
int main()
{
    int n,m;
    scanf ("%d%d%d",&n,&m,&mo);
    build(1,n,1);
    while(m--)
    {
        int flag,x,y,c;
        scanf ("%d",&flag);
        if(flag==1)
        {
            scanf ("%d%d%d",&x,&y,&c);
            mul(x,y,c,1,n,1);
        }
        else if(flag==2)
        {
            scanf ("%d%d%d",&x,&y,&c);
            add(x,y,c,1,n,1);
        }
        else
        {
            scanf ("%d%d",&x,&y);
            printf("%lld\n",ask_sum(x,y,1,n,1));
        }
    }
    return 0;
}

就这样。

2018.06.16

今天心血来潮准备打个线段树模板玩，结果调了TM一个多小时...

一交先是RE(我明明开足了4倍空间)，然后又T(你在打我脸)，最后加个快读终于A了，气死我了。

#include <cstdio>
typedef long long LL;
const int N = 100010;

LL MO;

inline void Read(LL &x) {
    x = 0;
    char c = getchar();
    while(c < '0' || c > '9') {
        c = getchar();
    }
    while(c <= '9' && c >= '0') {
        x = (x << 3) + (x << 1) + c - 48;
        c = getchar();
    }
    return;
}

struct SegmentTree {
    LL sum[N << 2], tag1[N << 2], tag2[N << 2];
    int l[N << 2], r[N << 2];

    inline void update(int o) {
        sum[o] = sum[o << 1] + sum[o << 1 | 1] % MO;
        return;
    }
    inline void pushdown(int o) {
        if(!tag2[o] && tag1[o] == 1) {
            return;
        }

        int ls = o << 1;
        int rs = ls | 1;
        int mid = (l[o] + r[o]) >> 1;

        tag1[ls] = tag1[ls] * tag1[o] % MO;
        tag2[ls] = (tag2[ls] * tag1[o] % MO + tag2[o]) % MO;
        sum[ls] = (sum[ls] * tag1[o] % MO + tag2[o] * (mid - l[o] + 1) % MO) % MO;

        tag1[rs] = tag1[rs] * tag1[o] % MO;
        tag2[rs] = (tag2[rs] * tag1[o] % MO + tag2[o]) % MO;
        sum[rs] = (sum[rs] * tag1[o] % MO + tag2[o] * (r[o] - mid) % MO) % MO;

        tag1[o] = 1;
        tag2[o] = 0;
        return;
    }
    void build(int o, int _l, int _r) {
        l[o] = _l;
        r[o] = _r;
        tag1[o] = 1;
        tag2[o] = 0;
        if(_l == _r) {
            Read(sum[o]);
            sum[o] %= MO;
            return;
        }
        int ls = o << 1;
        int rs = ls | 1;
        int mid = (_l + _r) >> 1;
        build(ls, _l, mid);
        build(rs, mid + 1, _r);
        update(o);
        return;
    }
    void add(int L, int R, int o, LL v) {
        if(L <= l[o] && r[o] <= R) {
            tag2[o] = (v + tag2[o]) % MO;
            sum[o] = (sum[o] + v * (r[o] - l[o] + 1) % MO) % MO;
            return;
        }
        if(r[o] < L || R < l[o]) {
            return;
        }
        
        pushdown(o);
        int ls = o << 1;
        int rs = ls | 1;
        int mid = (l[o] + r[o]) >> 1;

        if(L <= mid) {
            add(L, R, ls, v);
        }
        if(mid < R) {
            add(L, R, rs, v);
        }
        update(o);
        return;
    }
    void mul(int L, int R, int o, LL v) {
        if(L <= l[o] && r[o] <= R) {
            tag1[o] = v * tag1[o] % MO;
            tag2[o] = v * tag2[o] % MO;
            sum[o] = (sum[o] * v) % MO;
            return;
        }
        if(r[o] < L || R < l[o]) {
            return;
        }
        
        pushdown(o);
        int ls = o << 1;
        int rs = ls | 1;
        int mid = (l[o] + r[o]) >> 1;

        if(L <= mid) {
            mul(L, R, ls, v);
        }
        if(mid < R) {
            mul(L, R, rs, v);
        }
        update(o);
        return;
    }
    LL ask(int L, int R, int o) {
        if(L <= l[o] && r[o] <= R) {
            return sum[o];
        }
        if(r[o] < L || R < l[o]) {
            return 0;
        }
        
        pushdown(o);
        int ls = o << 1;
        int rs = ls | 1;
        int mid = (l[o] + r[o]) >> 1;

        LL ans = (ask(L, R, ls) + ask(L, R, rs)) % MO;
        return ans;
    }
}SegT;

int main() {
    int n, m;
    scanf("%d%d%lld", &n, &m, &MO);
    SegT.build(1, 1, n);
    for(int i = 1, x, y, f; i <= m; i++) {
        scanf("%d%d%d", &f, &x, &y);
        if(f == 3) {
            printf("%lld\n", SegT.ask(x, y, 1));
        }
        else {
            LL k;
            scanf("%lld", &k);
            if(f == 1) {
                SegT.mul(x, y, 1, k);
            }
            else {
                SegT.add(x, y, 1, k);
            }
        }
    }

    return 0;
}