Color the ball
Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 21795 Accepted Submission(s): 10552
Problem Description
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,输入结束。
当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
Author
8600
Source
中文题意不做解释了。
法一:
用的是线段树,就是区间更新和单点查询,就是单点查询每个都要查询到,感觉效率有点慢。
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <map>
#include <algorithm>
#include <math.h>
using namespace std;
#define rt x<<1|1
#define lt x<<1
#define LL long long
const int N = 100000*5;
int num[N];
int Add[N];
void pushdown(int x)
{
Add[lt]+=Add[x];
Add[rt]+=Add[x];
num[lt]+=Add[x];
num[rt]+=Add[x];
Add[x]=0;
}
void add(int x,int l,int r,int ll,int rr)
{
///if(Add[x]) pushdown(x);
if(ll<=l&&rr>=r)
{
num[x]++;
Add[x]++;
}
else
{
int mid=(l+r)>>1;
if(ll<=mid) add(lt,l,mid,ll,rr);
if(mid<rr) add(rt,mid+1,r,ll,rr);
num[x]=num[lt]+num[rt];
}
}
int query(int x,int l,int r,int i)
{
if(Add[x]) pushdown(x);
int ans=0;
if(l==r&&l==i)
{
ans+=num[x];
}
else
{
int mid=(l+r)>>1;
if(i<=mid) ans+=query(lt,l,mid,i);
if(mid<i) ans+=query(rt,mid+1,r,i);
}
return ans;
}
int main()
{
int n;
while(~scanf("%d",&n)&&n)
{
int l,r;
memset(num,0,sizeof num);
memset(Add,0,sizeof Add);
for(int i=1;i<=n;i++)
{
scanf("%d %d",&l,&r);
add(1,1,n,l,r);
}
for(int i=1;i<=n;i++)
{
if(i!=1) printf(" ");
printf("%d",query(1,1,n,i));
}
printf("\n");
}
}
法二:
树状数组,把问题当作单点更新,区间查询。
遇到左端点,左端点所在的+1,右端点+1所在的-1。查询点就直接变成了查询1-x区间上的值。这个思维就和法一完全不一样了。
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <map>
#include <algorithm>
#include <math.h>
using namespace std;
#define rt x<<1|1
#define lt x<<1
#define LL long long
const int N = 100000*5;
int n;
int num[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,int c)
{
while(x<=n)
{
num[x]+=c;
x+=lowbit(x);
}
}
int query(int x)
{
int ans=0;
while(x>=1)
{
ans+=num[x];
x-=lowbit(x);
}
return ans;
}
int main()
{
while(~scanf("%d",&n)&&n)
{
memset(num,0,sizeof num);
int l,r;
for(int i=1;i<=n;i++)
{
scanf("%d %d",&l,&r);
add(l,1);
add(r+1,-1);
}
for(int i=1;i<=n;i++)
{
printf("%d",query(i));
if(i==n) printf("\n");
else printf(" ");
}
}
}