树状数组例题

A 单点修改，区间查询

A. 单点修改，区间查询 [ Discussion ]
Description
给定数列a[1],a[2],…,a[n]，你需要依次进行 q个操作，操作有两类：

1 i x：给定i,x，将a[i]加上x；

2 l r：给定l,r，求 ∑ri=la[i]的值（换言之，求a[l]+a[l+1]+⋯+a[r]的值）

Input
第一行包含2个正整数n,q，表示数列长度个数，保证a≤n,q≤106
第二行n个整数a[1],a[2],…,a[n]，表示初始数列，保证|a[i]|≤106
接下来q行，每行一个操作，为下列两种之一：

1 i x：给定i,x，将a[i]加上x；

2 l r：给定l,r，求 ∑ri=1a[i]的值；

保证1≤l≤r≤n， |x|≤106
Output
对于每个 2 l r 操作输出一行，每行有一个整数，表示所求的结果。

Samples
Input Copy
3 2
1 2 3
1 2 0
2 1 3
Output
6
Hint
对于所有数据，1≤n,q≤106，|a[i]|≤106，1≤l≤r≤n，|x|≤106

#include<bits/stdc++.h>;
#include<stdio.h>
#include<string.h>
using namespace std;
typedef long long ll;
ll read(){ll res = 0, ch, flag = 0;if((ch = getchar()) == '-')flag = 1;else if(ch >= '0' && ch <= '9')res = ch - '0';while((ch = getchar()) >= '0' && ch <= '9' )res = res * 10 + ch - '0';return flag ? -res : res;}
const int maxn =1e6+199 ;
//ll sum=0,maxa=-1;
ll n,m,k;
ll a[maxn]={0};
char b[maxn];
ll dis[maxn];
map<int,int>mp;
void add(int x,int s)
{
    for(int i=x;i<=n;i+= ( i & -i) )  a[i]+=s;
}
ll sum1(int x)
{
    ll k=0;
    for(int i=x;i>=1;i-=( i & -i) )   k+=a[i];
    return k;
}
int main(){

     n=read();
  int t=read();
    for(int i=1;i<=n;i++)
    {
        int x=read();
        add(i,x);
    }
    while(t--)
    {
        k=read();

        if(k==1)
        {
            int d=read();
            int s=read();
            add(d,s);
        }
        if(k==2)
        {
            int z=read();
            int y=read();

            printf("%lld\n",sum1(y)-sum1(z-1));
        }

    }


    return 0;

}

B. 区间修改，单点查询 [ Discussion ]
Description
给定数列a[1],a[2],…,a[n] ，你需要依次进行 q 个操作，操作有两类：

1 l r x：给定 l,r,x，对于所有i∈[l,r]，将 a[i] 加上 x（换言之，将 a[l],a[l+1],…,a[r] 分别加上 x）；
2 i：给定 i ，求 a[i] 的值。
Input
第一行包含 2 个正整数 n,q，表示数列长度和询问个数。保证1≤n,q≤106 。
第二行 n 个整数 a[1],a[2],…,a[n]，表示初始数列。保证|a[i]|≤106 。
接下来 q 行，每行一个操作，为以下两种之一：

1 l r x：对于所有 i∈[l,r]，将a[i]加上x；
2 i：给定 i，求 a[i] 的值。
保证 1≤l≤r≤106, |x|≤106。

Output
对于每个 2 i 操作，输出一行，每行有一个整数，表示所求的结果。

Samples
Input Copy
3 2
1 2 3
1 1 3 0
2 2
Output
2
Hint
对于所有数据，1≤n,q≤106 , |a[i]|≤106， 1≤l≤r≤n, |x|≤106
Discussions
No more discussions

#include<bits/stdc++.h>;
#include<stdio.h>
#include<string.h>
using namespace std;
typedef long long ll;
ll read(){ll res = 0, ch, flag = 0;if((ch = getchar()) == '-')flag = 1;else if(ch >= '0' && ch <= '9')res = ch - '0';while((ch = getchar()) >= '0' && ch <= '9' )res = res * 10 + ch - '0';return flag ? -res : res;}
const int maxn =1e6+199 ;
//ll sum=0,maxa=-1;
ll n,m,k;
priority_queue<int, vector< int >,less< int > > q;
priority_queue<int, vector< int >,greater< int > > p;
ll a[maxn]={0};
char b[maxn];
ll dis[maxn];
map<int,int>mp;
void add(int x,int s)
{
    for(int i=x;i<=n;i+= ( i & -i) )  a[i]+=s;
}
ll sum1(int x)
{
    ll k=0;
    for(int i=x;i>=1;i-=( i & -i) )   k+=a[i];
    return k;
}
int main(){

     int y = 0;

     n=read();
     int t=read();
    for(int i=1;i<=n;i++)
    {
        int x=read();
        add(i,x-y);
        y=x;
    }
    while(t--)
    {
        k=read();

        if(k==1)
        {
            int z=read();
            int y=read();
            int s=read();
            add(z,s);
            add(y+1,-s);
        }
        if(k==2)
        {
            int z=read();


            printf("%lld\n",sum1(z));
        }

    }


    return 0;

}

C. 区间修改，区间查询 [ Discussion ]
Description
给定数列 a[1],a[2],…,a[n]，你需要依次进行q个操作，操作有两类：

1 l r x：给定l,r,x，对于所有的i∈[l,r]，将a[i]加上x（换言之，将a[l],a[l+1],…,a[r] 分别加上x）
2 l r：给定l,r，求∑ri=la[i]的值（换言之，求a[l]+a[l+1]+…+a[r]的值）
Input
第一行包含2个正整数n,q，表示数列长度和询问个数。保证1≤n,q≤106；

第二行n个整数a[1],a[2],…,a[n]，表示初始数列。保证|a[i]|≤106。

接下来q行，每行一个操作，为以下两种之一：

1 l r x：对于所有的i∈[l,r]，将a[i]加上x；

2 l r：输出∑ri=la[i]的值；

Output
对于每个 2 l r 操作，输出一行，每行有一个整数，表示所求的结果。

Samples
Input Copy
5 10
2 6 6 1 1
2 1 4
1 2 5 10
2 1 3
2 2 3
1 2 2 8
1 2 3 7
1 4 4 10
2 1 2
1 4 5 6
2 3 4
Output
15
34
32
33
50
Hint
对于所有数据，1≤n,q≤106，|a[i]|≤106 ，1≤l≤r≤n，|x[i]|≤106 。

这个代码不知哪错了 先留着看看

#include<bits/stdc++.h>;
#include<stdio.h>
#include<string.h>
using namespace std;
typedef long long ll;
ll read(){ll res = 0, ch, flag = 0;if((ch = getchar()) == '-')flag = 1;else if(ch >= '0' && ch <= '9')res = ch - '0';while((ch = getchar()) >= '0' && ch <= '9' )res = res * 10 + ch - '0';return flag ? -res : res;}
const int maxn =1e6+199 ;
//ll sum=0,maxa=-1;
ll n,m,k;
priority_queue<int, vector< int >,less< int > > q;
priority_queue<int, vector< int >,greater< int > > p;
ll a[maxn]={0};
char b[maxn];
ll dis[maxn];
map<int,int>mp;
ll sum[maxn];

void add(ll a[],int x,int s)
{
    for(int i=x;i<=n;i+= ( i & -i) )  a[i]+=s;
}
ll sum1(int x)
{
    ll k=0;

    for(int i=x;i>=1;i-=( i & -i) )   k+=x*a[i]-sum[i];

    return k;
}
int main(){

     int y = 0;

     n=read();
     int t=read();
    for(int i=1;i<=n;i++)
    {
        int x=read();
        add(a,i,x-y);
        add(sum,i,(i-1)*(x-y));
        y=x;
    }
    while(t--)
    {
        k=read();

        if(k==1)
        {
            int z=read();
            int y=read();
            int s=read();
            add(a,z,s);
            add(a,y+1,-s);
            add(sum,z,s*(z-1));
            add(sum,y+1,-s*y);
        }
        if(k==2)
        {
            int z=read();
            int y=read();
            printf("%lld\n",sum1(y)-sum1(z-1));
        }

    }


    return 0;