题目-一个简单的整数问题-树状数组与差分的结合

给定长度为N的数列A,然后输入M行操作指令。
第一类指令形如“C l r d”,表示把数列中第l~r个数都加d。
第二类指令形如“Q X”,表示询问数列中第x个数的值。
对于每个询问,输出一个整数表示答案。
输入格式
第一行包含两个整数N和M。
第二行包含N个整数A[i]。
接下来M行表示M条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤1051≤N,M≤105,

|d|≤10000|d|≤10000,

|A[i]|≤1000000000|A[i]|≤1000000000
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

输出样例:
4
1
2
5
思路:
树状数组可以把某一个数+x,求一段区间的和
而这道题是让我们在某一个区间加x,求一个数的值
所以这道题可以采取树状数组与差分结合的思路,树状数组存储的是原数组的差分值。
同时我们观察到数据范围较大,所以用long long来得到答案

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100010;
int n,m;
typedef long long LL;
int a[N];
LL tr[N];
int lowbit(int x){
 return x&-x;
}
void add(int x,int c){
 for(int i=x;i<n;i+=lowbit(i))    tr[i]+=c;
}
LL sum(int x){
 LL res=0;
 for(int i=x;i;i-=lowbit(i))      res+=tr[i];
 return  res;
}
int main(){
 scanf("%d%d",&n,&m);
 for(int i=1;i<=n;i++)    scanf("%d",&a[i]);
 for(int i=1;i<=n;i++)    add(i,a[i]-a[i-1]);//加进去的是a[i]的差分数组,方便进行后面的操作 
    while(m--){
     char op[2];
     int l,r,d;
     scanf("%s%d",op,&l);
     if(*op=='C'){
      scanf("%d%d",&r,&d);
      add(l,d);
      add(r+1,-d);
  }
  else{
   printf("%lld\n",sum(l));
  }
 }
 return 0;
}
发布了29 篇原创文章 · 获赞 24 · 访问量 3672

猜你喜欢

转载自blog.csdn.net/qq_45772483/article/details/104301686