Color the ball
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<iostream>
#include<cstring>
#include<math.h>
#include<stdlib.h>
#include<cstring>
#include<cstdio>
#include<utility>
#include<algorithm>
#include<map>
#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<<query(i,i)<<" ";
}
cout<<endl;
}
return 0;
}
正解:
#include<iostream>
#include<cstring>
#include<math.h>
#include<stdlib.h>
#include<cstring>
#include<cstdio>
#include<utility>
#include<algorithm>
#include<map>
#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<n;++i){
cout<<getsum(i)<<" ";
}
cout<<getsum(n)<<endl;
}
return 0;
}
差分+树状数组
#include<iostream>
#include<cstring>
#include<math.h>
#include<stdlib.h>
#include<cstring>
#include<cstdio>
#include<utility>
#include<algorithm>
#include<map>
#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<n;++i){
cout<<getsum(i)<<" ";
}
cout<<getsum(n)<<endl;
}
return 0;
}