牛客-区间 (interval)

版权声明：版权归Ordinarv所有 https://blog.csdn.net/ordinarv/article/details/85124993

链接：https://ac.nowcoder.com/acm/problem/16722
来源：牛客网
 

题目描述

    Apojacsleam喜欢数组。

    他现在有一个n个元素的数组a，而他要对a[L]-a[R]进行M次操作：

        操作一：将a[L]-a[R]内的元素都加上P

        操作二：将a[L]-a[R]内的元素都减去P

    最后询问a[l]-a[r]内的元素之和？

    请认真看题干及输入描述。

输入描述:

输入共M+3行：

第一行两个数，n，M，意义如“题目描述”

第二行n个数，描述数组。

第3-M+2行，共M行，每行四个数，q，L，R，P，若q为1则表示执行操作2，否则为执行操作1

第4行，两个正整数l，r

输出描述:

一个正整数，为a[l]-a[r]内的元素之和

示例1

输入

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5 5
1 2 3 6
0 2 5 5 
0 2 5 8
1 4 9 6
2 7

输出

23

说明

---

分析：

树状数组区间求和

---

AC Code

#include<iostream>
#include<cstring>
#include<queue>
#include<cstdio>
using namespace std;
typedef long long ll;
ll c[1000005];
int n,m;
int lowbit(int x)
{
    return x&(-x);
}
void add(int pos,int x)
{
    while(pos<=n)
    {
        c[pos]+=x;
        pos+=lowbit(pos);
    }
}
ll query(int pos)
{
    ll res=0;
    while(pos>0)
    {
        res+=c[pos];
        pos-=lowbit(pos);
    }
    return res;
 }
int main()
{
    scanf("%d%d",&n,&m);
    int x=0,y;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&y);
        add(i,y-x);
        x=y;
    }
    int opt;
    ll k;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d%d",&x,&y,&k);
            add(x,-k);
            add(y+1,k);
        }
        else
        {
            scanf("%d%d%d",&x,&y,&k);
            add(x,k);
            add(y+1,-k);
        }
    }
  
    int s,t;
    ll sum=0;
//    for(int i=1;i<=n;i++)
//        cout<<query(i)<<' '; cout<<endl;
    cin>>s>>t;
    for(int i=s;i<=t;i++)
        sum+=query(i);
    cout<<sum<<endl;
    return 0;
 }