Color the ball
Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 5328 Accepted Submission(s): 2839
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,输入结束。
Output
每个测试实例输出一行,包括N个整数,第I个数代表第I个气球总共被涂色的次数。
Sample Input
3
1 1
2 2
3 3
3
1 1
1 2
1 3
0
Sample Output
Author
8600
Source
HDU 2006-12 Programming Contest
Recommend
LL
我的第一道线段树题。一次ac感觉很爽。线段树真是个神奇的东西。
#include <stdio.h>
struct tree
{
int l,r;
int cnt;
} t[300100];//不知道为什么这么大。。。
void btree(int s,int e,int k)//构造线段树
{
int mid;
t[k].l=s;
t[k].r=e;
t[k].cnt=0;
mid=(t[k].l+t[k].r)/2;
if(s==e)
return;
btree(s,mid,2*k);
btree(mid+1,e,2*k+1);
}
void inst(int s,int e,int k)//更新线段树
{
int mid;
if(t[k].l==s&&t[k].r==e)
{
t[k].cnt++;
return;
}
mid=(t[k].l+t[k].r)/2;
if(e<=mid)
inst(s,e,2*k);
else if(s>mid)
inst(s,e,2*k+1);
else
{
inst(s,mid,2*k);
inst(mid+1,e,2*k+1);
}
}
int findx(int k,int x)//查找线段树
{
int sum,mid;
sum=t[k].cnt;
if(t[k].l==t[k].r)
return sum;
mid=(t[k].l+t[k].r)/2;
if(x>mid)
sum+=findx(2*k+1,x);
else
sum+=findx(2*k,x);
return sum;
}
int main()
{
int n,a,b,i;
while(scanf("%d",&n),n)
{
btree(1,n,1);
for(i=1;i<=n;i++)
{
scanf("%d%d",&a,&b);
inst(a,b,1);
}
for(i=1;i<n;i++)
printf("%d ",findx(1,i));
printf("%d\n",findx(1,i));
}
return 0;
}
树状数组效率是令人欣慰的。
#include <iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
int n;
int c[100100];
int lowbit(int x)
{
return x&(-x);
}
void update(int x,int chan)
{
while(x>0)
{
c[x]+=chan;
x-=lowbit(x);
}
}
int qu(int x)
{
int s=0;
while(x<=n)
{
s+=c[x];
x+=lowbit(x);
}
return s;
}
int main()
{
int a,b,i;
while(scanf("%d",&n),n)
{
memset(c,0,(n+1)*sizeof(int));
for(i=1; i<=n; i++)
{
scanf("%d%d",&a,&b);
update(b,1);
update(a-1,-1);
}
for(i=1; i<n; i++)
printf("%d ",qu(i));
printf("%d\n",qu(i));
}
return 0;
}