zoukankan      html  css  js  c++  java
  • 树状数组的区间查询与区间修改

    用c1存表示a[i]-a[i-1]的树状数组
    用c2存(i-1)*c1[i]

    sum[x]=c1[1]+c1[1]+c1[2]+c1[1]+c1[2]+c1[3]+...+c1[1]+...+c1[x]
              =x*(c1[1]+c1[2]+c1[3]+…+c1[x])-(0*c1[1]+1*c1[2]+2*c1[3]+…+(x-1)*c1[x])
    
    #include<iostream>
    #include<cstdio>
    using namespace std;
    
    const long long MAXN=1e6+5;
    
    long long c1[MAXN],c2[MAXN];
    long long n,m;
    
    long long lowbit(long long x){return x&(-x);}
    void upd(long long f,long long t,long long w);
    long long ask(long long p);
    
    int main(){
    	scanf("%d%d",&n,&m);
    	for(long long i=1;i<=n;++i){
    		long long t;
    		cin>>t;
    		upd(i,i,t);
    	}
    	for(long long i=1;i<=m;++i){
    		char q;
    		cin>>q;
    		if(q=='Q'){
    			long long t1,t2;
    			cin>>t1>>t2;
    			long long ans=ask(t2)-ask(t1-1);
    			cout<<ans<<'
    ';
    		}
    		else{
    			long long t1,t2,t3;
    			cin>>t1>>t2>>t3;
    			upd(t1,t2,t3);
    		}
    	}
    	return 0;
    }
    
    void upd(long long f,long long t,long long w){
    	long long add1=(f-1)*w,add2=t*w;
    	for(long long i=f;i<=n;i+=lowbit(i)){
    		c1[i]+=w;
    		c2[i]+=add1;
    	}
    	for(long long i=t+1;i<=n;i+=lowbit(i)){
    		c1[i]-=w;
    		c2[i]-=add2;
    	}
    }
    
    long long ask(long long p){
    	long long ans1=0,ans2=0;
    	for(long long i=p;i>0;i-=lowbit(i)){
    		ans1+=c1[i];
    		ans2+=c2[i];
    	}
    	return p*ans1-ans2;
    }
    
  • 相关阅读:
    C++ 项目和资源导引
    C++ 类再探
    C++ 语句函数再探
    leetcode-174. Dungeon Game 地下城游戏
    34. Find First and Last Position of Element in Sorted Array + 二分
    leetcode-27. Remove Element删除元素
    git 使用入门
    MySQL数据库的启动与停止
    C++类型转换
    C++ 获取对象类型
  • 原文地址:https://www.cnblogs.com/buringstraw/p/9984328.html
Copyright © 2011-2022 走看看