树状数组模板

Color the ball

 HDU - 1556 

N个气球排成一排,从左到右依次编号为1,2,3....N.每次给定2个整数a b(a <= b),lele便为骑上他的“小飞鸽"牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第I个气球已经涂过几次颜色了,你能帮他算出每个气球被涂过几次颜色吗?

Input

每个测试实例第一行为一个整数N,(N <= 100000).接下来的N行,每行包括2个整数a b(1 <= a <= b <= N)。 
当N = 0,输入结束。

Output

每个测试实例输出一行,包括N个整数,第I个数代表第I个气球总共被涂色的次数。

Sample Input

3
1 1
2 2
3 3
3
1 1
1 2
1 3
0

Sample Output

1 1 1
3 2 1

思路不难,就是线段树||差分数组||普通树状数组||树状数组+差分的模板题

每次更新时区间各点+1,最后输出每个元素的值,总结来说就是区间更新+单点查询

 

TLE版本,单点修改,肯定超时,因为需要每次遍历左端点到右端点去更新,无法保持O(logn)的时间复杂度

普通正解:update数组表示从1到x每个数更新的次数,所以我们只需要update(L-1,-1)和update(R,1)表示[L,R]被更新的次数,相当于将[1,L-1]这个区间-1,再将[1,R]+1

差分+树状数组:在普通正解基础上,在每次修改区间时,利用差分数组的性质,只更新左端点L和R+1的位置

TLE:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#define lowbit(i) ((i)&(-i))
using namespace std;
typedef long long ll; 
inline int read(){
    int X=0,w=0;char ch=0;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
/*------------------------------------------------------------------------*/
const int maxn=1e5+10; 
ll c[maxn];//树状数组
int n;
void update(int x,int v){
	for(int i=x;i<=n;i+=lowbit(i)){
		c[i]+=v;
	}
} 
ll getsum(int x){//前x个整数之和
	ll sum=0;
	for(int i=x;i>0;i-=lowbit(i)){
		sum+=c[i];
	}
	return sum;
}
ll query(int l,int r){
	return getsum(r)-getsum(l-1);
}
int main( )
{	
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    //freopen("a.txt","r",stdin);
    //freopen("a.txt","w",stdout);
    
    while(cin>>n){
    	if(n==0)break;
    	memset(c,0,sizeof(c));
    	
    	int l,r;
    	for(int i=1;i<=3;++i){
    		
    		cin>>l>>r;
    		for(int j=l;j<=r;++j){
    			update(j,1);
    		}
    		
    	}
    	
    	for(int i=1;i<=n;++i){
    		cout<

正解:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#define lowbit(i) ((i)&(-i))
using namespace std;
typedef long long ll; 
inline int read(){
    int X=0,w=0;char ch=0;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
/*------------------------------------------------------------------------*/
const int maxn=1e5+10; 
ll c[maxn];//树状数组
int n;
void update(int x,int v){
	for(int i=x;i>0;i-=lowbit(i)){
		c[i]+=v;
	}
} 
ll getsum(int x){//求第i个整数的值 
	ll sum=0;
	for(int i=x;i<=n;i+=lowbit(i)){
		sum+=c[i];
	}
	return sum;
}
ll query(int l,int r){
	return getsum(r)-getsum(l-1);
}
int main( )
{	
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    //freopen("a.txt","r",stdin);
    //freopen("a.txt","w",stdout);
    
    while(cin>>n){
    	if(n==0)break;
    	memset(c,0,sizeof(c));
    	
    	int l,r;
    	for(int i=1;i<=n;++i){
    		
    		cin>>l>>r;
    		
    		update(r,1);
    		update(l-1,-1);
    		
    	}
    	
    	for(int i=1;i

差分+树状数组

#include
#include
#include
#include
#include
#include
#include
#include
#include
#define lowbit(i) ((i)&(-i))
using namespace std;
typedef long long ll; 
inline int read(){
    int X=0,w=0;char ch=0;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
/*------------------------------------------------------------------------*/
const int maxn=1e5+10; 
ll c[maxn];//树状数组
int n;
void update(int x,int v){
	for(int i=x;i<=n;i+=lowbit(i)){
		c[i]+=v;
	}
} 
ll getsum(int x){
	ll sum=0;
	for(int i=x;i>0;i-=lowbit(i)){//前x个整数之和 
		sum+=c[i];
	}
	return sum;
}
ll query(int l,int r){
	return getsum(r)-getsum(l-1);
}
int main( )
{	
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    //freopen("a.txt","r",stdin);
    //freopen("a.txt","w",stdout);
    
    while(cin>>n){
    	if(n==0)break;
    	memset(c,0,sizeof(c));
    	
    	int l,r;
    	for(int i=1;i<=n;++i){
    		
    		cin>>l>>r;
    		
    		//update(r,1);
    		//update(l-1,-1);
    		update(l,1);
    		update(r+1,-1);
    		 
    	}
    	/*
		差分数组的性质:差分数组的前x个数的和就是x的值 
		*/ 
    	for(int i=1;i

 

你可能感兴趣的:(树状数组)