WA声一片,用树状数组写的,求调(玄关))
查看原帖
WA声一片,用树状数组写的,求调(玄关))
815011
WangYao0126楼主2024/1/29 11:59
#include<bits/stdc++.h>
using namespace std;
int n,m;
long long fw[100005],fw1[100005],a[100005];
void add(int p,int v){
	for(long long q=p*v;p<=n;p+=(p&-p)){
		fw[p]+=v;
		fw1[p]=q;
	}
    return ;
}
int query(int p){
	long long res=0;
	for(int i=p;i!=0;i-=(i&-i))
		res+=fw[i]*(p+1)-fw1[i];
	return res;
}
int query1(int l,int r){
	return query(r)-query(l-1);
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=n;i>=1;i--)
		add(i,a[i]-=a[i-1]);
	for(int i=1;i<=m;i++){
		int type;
		cin>>type;
		if(type==1){
			int l,r,k;
			cin>>l>>r>>k;
			add(l,k);
			add(r+1,-k);
		} 
		else{
			int l,r;
			cin>>l>>r;
			cout<<query1(l,r)<<'\n';
		}
	}
	return 0;
}
2024/1/29 11:59
加载中...