JZ初中OJ 1567. [GDKOI]簡單題

input:simple.in output:simple.out

時間限制:  1000 ms  空間限制: 262144 KB  具體限制  

題目描述

    給你N個整數A1,A2,….,AN，你可以進行兩種操作：一種把一給定區間的每個數是增加一個值；另一種是計算某一給定區間的數字的和。

 

輸入

    第一行包含兩個整數N和Q.1<=N,Q<=100000

第二行包含N個整數，是A1,A 2,….,AN的初始值。-1000000000<=Ai<=1000000000

接下來Q行，描述操作。

“C a b c”表示把Aa,Aa+1,….Ab都增加c，-10000<=c<=10000。

“Q a b”表示詢問Aa,Aa+1,….,Ab的和。

輸出

對於每個詢問輸出對應的和，每行一個。

 

樣例輸入

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

 

數據范圍限制

【限制】

30%數據 N,Q<=10000

#include<bits/stdc++.h>
using namespace std;
long long n,q,d,s,c1[100001],sum1[100001],c2[100001],a[100001];
char w;
long long lowbit(long long x)
{
    return x&(-x);
}
long long sum(long long c[],long long x)
{
    long long ret=0;
    while(x>0)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
void add(long long c[],long long x,long long y)
{
    while(x<=n)
    {
        c[x]+=y;
        x+=lowbit(x);
    }
}
int main()
{
    freopen("simple.in","r",stdin);
    freopen("simple.out","w",stdout);
    cin>>n>>q;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        sum1[i]=sum1[i-1]+a[i];
    }
    for(int j=1;j<=q;j++)
    {
        cin>>w;
        cin>>s>>d;
        if(int(w)==int('Q'))
        {
            long long suml=sum1[s-1]+s*sum(c1,s-1)-sum(c2,s-1);
            long long sumd=sum1[d]+(d+1)*sum(c1,d)-sum(c2,d);
            cout<<sumd-suml<<endl; 
        }
        else 
        {
            long long cost;
            cin>>cost;
            add(c1,s,cost);
            add(c1,d+1,-cost);
            add(c2,s,s*cost);
            add(c2,d+1,-(d+1)*cost);
        }
    }
    return 0;
}