牛客小白月赛5 I-区间 树状数组

https://www.nowcoder.com/acm/contest/135/I

虽然正解不是树状数组,但是刚学了树状数组就拿来用了,也能过,改成区间更新,单点求值的

注意:数据范围 maxn

          答案范围 int会溢出 改成long long

          c[i]下标从1开始,不要弄错了

#include<cstdio>
#include<algorithm>
#include<cstring>
#define CLR(x) memset(x,0,sizeof(x))
#define ll long long int
#define PI acos(-1.0)
#define db double
#define mod 1e9+7
using namespace std;
const int maxn = 1e6+6;
const int inf = 1e9;
const db eps = 1e-6;
int c[maxn];
int lisan[maxn];
int s[maxn];
int n,k;
int lowbit(int x){
	return x&(-x);
}
void update(int i, int value){
	while(i >0){
		c[i] += value;
		i -= lowbit(i);
	}
}
int sum(int i){
	int s = 0;
	while(i <=n){
		s += c[i];
		i += lowbit(i);
	}
	return s;
}
int main(){
	int m,a;
	scanf("%d%d",&n,&m);
	for(int i = 1; i<= n;i++){
		scanf("%d",&a);
		update(i,a);
		update(i-1,-a);
	}
	int q,l,r,p;
	for(int i = 0;i<m;i++){
		scanf("%d%d%d%d",&q,&l,&r,&p);
		if(q!=1){
			update(r,p);
			update(l-1,-p);
		}
		else{
			update(r,-p);
			update(l-1,p);
		}
	}
	int L,R;
	scanf("%d%d",&L,&R);
	ll ans = 0;
	for(int i = L;i<=R;i++)
		ans+=sum(i);
	printf("%lld\n",ans);
}

猜你喜欢

转载自blog.csdn.net/qq_41156122/article/details/81179520
今日推荐