樹狀數組_應用的代碼

empty#include 
#include 
#include 
#include 

using namespace std;
const int N = 500005;

struct Node
{
    int v,order;
};

int n;
int c[N];
int aa[N];    //離散化後的數組
Node a[N];    //樹狀數組

int Lowbit(int x)
{
    return x & (-x);
}

void Update(int t,int val)
{
    for(int i=t; i<=n; i+=Lowbit(i))
        c[i] += val;
}

int getSum(int x)
{
    int ans=0;
    for(int i=x; i>0; i-=Lowbit(i))
        ans += c[i];
    return ans;
}

bool cmp(Node a,Node b)
{
    return a.v
#include
using namespace std;
#define N 10005
long long c1[N];//維護delta【i】的前綴和 
long long c2[N];//維護delta[i]*i的前綴和
long long sum[N];//sum[i]=A[1]+..+A[i],記錄A[i]的前綴和
long long A[N];
int n;
//long long array[N];//樹狀數組的引用
int lowbit(int x){
	return x&-x;
}
long long update (long long * array,int i,long long d){
	while(i<=n){
		array[i]+=d;
		i+=lowbit(i); 
	}
} 
long long get_sum(long long *array,int i){
	long long sum_;
	sum_=0;
	while(i>0){
		sum_+=array[i];
		i-=lowbit(i);
	}
	return sum_;
}
int main(){
	int t;
	cin>>t;
	while(t--){
		char operation;
		int s,t;
		long long array[N];
		memset(array,0,sizeof(array));
		int query_times;
		cin>>n>>query_times;
		for(int i=1;i<=n;i++){
			cin>>A[i];
		}
		for(int i=1;i<=n;i++){
			sum[i]=sum[i-1]+A[i]; 
		}
		 while(query_times){
		 	getchar();
		 	cin>>operation>>s>>t;
		 	if(operation=='Q'){
		 		int ans=sum[t]-sum[s-1];
		 		//做更新的時候只需要像普通的區間更新
				 //這裏查詢的時候分成兩塊,一塊是基數,一開始輸入的數據,用sum[i]存儲他的前綴和
				 //然後是區間內的累積增加量且delta[i]對sum[x]的貢獻值爲delta[i]*(x+1-i)
		 		//由於sum是記入了前綴和,而delta是一個樹狀數組,我們更新他的時候是更新到最後一個父節點
				 //所以能把sun[x]影響的delta[i]的個數就是(x+1-i)個這麼多 
				 //化簡後 sum[x]=sum[x](基數) + (x+1)*segma(delta[i]) - segma(delta[i]*i) 
				 ans+=(t+1)*get_sum(c1,t)-get_sum(c2,t);
			    ans-=(s)*get_sum(c1,s)-get_sum(c2,s);
			    cout<>need_add;
			 	update(c1,s,need_add);
			 	update(c1,t+1,-need_add);
			 	update(c2,s,need_add*s);
			 	update(c2,t+1,-need_add*s);
			 }
		 }
	} 
} 
發佈了35 篇原創文章 · 獲贊 0 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章