cogs 1317. 数列操作C 区间修改 区间查询

1317. 数列操作C

★★★   输入文件：shuliec.in   输出文件：shuliec.out   简单对比
时间限制：1 s   内存限制：128 MB

【题目描述】

假设有一个长度为 n(n≤100000) 的数列 A，支持如下两种操作：

1. 将 Ai,Ai+1,…,Aj 的值均增加 d

2. 查询 As+As+1+⋯+At(s≤t) 的值。

根据操作要求进行正确操作并输出结果。

【输入格式】

第一行为一个正整数 n，表示数列的大小。

第二行有 n 个整数，表示数列 A 各项的初始值。

第三行为一个整数 m ，表示操作的个数。

下面是 m 行，每行描述一个操作：

ADD i j d（将 Ai,Ai+1,…,Aj(1≤i,j≤n) 的值均增加一个整数 d）

SUM s t（表示查询 As+⋯+At 的值）

【输出格式】

对于每一次询问，输出查询到的结果。

【样例输入】

4
1 4 2 3
3
SUM 1 3
ADD 2 2 50
SUM 2 3

【样例输出】

7
56

【提示】

所有答案小于 4611686018427387904

加强 10 组极限数据，未全部重测 by rvalue 2018.2.26

还是线段树而已

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=;
#define ll long long
ll a[maxn];
int n,m;
struct SegmentTree{
    int l,r;
    long long dat;
    long long lazy_tag;
}t[maxn<<];
void Pushdown(int p,int l,int r,int mid){
    t[p*].dat+=t[p].lazy_tag*(mid-l+);
    t[p*+].dat+=t[p].lazy_tag*(r-mid);
    t[p*].lazy_tag+=t[p].lazy_tag;
    t[p*+].lazy_tag+=t[p].lazy_tag;
    t[p].lazy_tag=;
}
void build(int p,int l,int r){
    t[p].l=l;t[p].r=r;
    if(l==r){
        t[p].dat=a[l];
        return;
    }
    int mid=(l+r)>>;
    build(p*,l,mid);
    build(p*+,mid+,r);
    t[p].dat=t[p*].dat+t[p*+].dat;
}
ll Get(int p,int l,int r,int s,int tt){
    if(s>r||tt<l) return ;
    if(s<=l&&r<=tt) return t[p].dat;
    int mid=(l+r)>>;
    Pushdown(p,l,r,mid);
    return  Get(p*,l,mid,s,tt)+ Get(p*+,mid+,r,s,tt);
}
void Add(int p,int l,int r,int s,int tt,ll x){
    if(s>r||tt<l) return;
    if(s<=l&&r<=tt){
        t[p].dat+=x*(r-l+);
        t[p].lazy_tag+=x;
        return;
    }
    int mid=(l+r)>>;
    Pushdown(p,l,r,mid);
    Add(p*,l,mid,s,tt,x);Add(p*+,mid+,r,s,tt,x);
    t[p].dat=t[p*].dat+t[p*+].dat;
}
int main()
{
    freopen("shuliec.in","r",stdin);freopen("shuliec.out","w",stdout);
    scanf("%d",&n);
    for(int i=;i<=n;i++) scanf("%lld",&a[i]);
    scanf("%d",&m);
    build(,,n);
    for(int i=;i<=m;i++){
        string s;cin>>s;
        if(s[]=='S'){
            int l,r;scanf("%d%d",&l,&r);
            printf("%lld\n",Get(,,n,l,r));
        }
        else{
            int s,t;
            ll d;
            scanf("%d%d%lld",&s,&t,&d);
            Add(,,n,s,t,d);
        }
    }

    return ;
 } 

还有一个

标记永久化线段树

#include <stdio.h>
#include <algorithm>
using namespace std;
const int maxn=;
#define LL long long
#define mid (l+r>>1)
#define ls (rt<<1)
#define rs (rt<<1|1)
#define lson ls,l,mid
#define rson rs,mid+1,r
int n,m,cnt,a[maxn];
LL sum[maxn<<],lazy[maxn<<];
void Build(int rt,int l,int r){
    if(l==r){
        sum[rt]=a[l];
        return;
    }
    Build(lson),Build(rson);
    sum[rt]=sum[ls]+sum[rs];
}
void Add(int rt,int l,int r,int s,int t,LL qx){
    if(s>r||t<l)return;
    if(s<=l&&r<=t){
        lazy[rt]+=qx;
        sum[rt]+=qx*(r-l+);
        return;
    }
    Add(lson,s,t,qx),Add(rson,s,t,qx);
    sum[rt]=sum[ls]+sum[rs]+lazy[rt]*(r-l+);
}
LL Get(int rt,int l,int r,int s,int t){
    if(s>r||t<l)return ;
    if(s<=l&&r<=t)return sum[rt];
    int ll=max(l,s),rr=min(r,t);
    return lazy[rt]*(rr-ll+)+Get(lson,s,t)+Get(rson,s,t);
}
int main(){
    freopen("shuliec.in","r",stdin);freopen("shuliec.out","w",stdout);
    scanf("%d",&n);
    for(int i=;i<=n;i++)scanf("%d",&a[i]);
    Build(,,n);
    scanf("%d",&m);
    for(int i=;i<m;i++){
        char s[];
        scanf("%s",s);
        if(s[]=='S'){
            int l,r;scanf("%d%d",&l,&r);
            printf("%lld\n",Get(,,n,l,r));
        }
        else{
            int l,r;LL x;scanf("%d%d%lld",&l,&r,&x);
            Add(,,n,l,r,x);
        }
    }
}

标记永久化就是在打标记的时候只需要打在上面就行了

不需要往下放