您的位置:首页 > 其它

H - Color the ball(树状数组)(区间更新)(求单点)(线段树)

2017-08-09 13:40 302 查看

H- 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
线段树解法:
#include<iostream>#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;#define maxn 100005int n;struct node{int left,right,val;}tree[maxn<<2];void build(int k,int l,int r){tree[k].left=l;tree[k].right=r;tree[k].val=0;if(l==r)return ;int mid=(l+r)>>1;build(k<<1,l,mid);build(k<<1|1,mid+1,r);} void update(int k,int l,int r){if(tree[k].left>=l&&tree[k].right<=r){tree[k].val++;return;}int mid=(tree[k].left+tree[k].right)>>1;if(r<=mid)return update(k<<1,l,r);else if(l>mid)return update(k<<1|1,l,r);else {update(k<<1,l,mid);update(k<<1|1,mid+1,r);}}int query(int k,int pos){if(tree[k].left==tree[k].right)return tree[
4000
k].val;if(tree[k].val){ tree[k<<1].val+=tree[k].val; tree[k<<1|1].val+=tree[k].val; tree[k].val=0;}int mid=(tree[k].left+tree[k].right)>>1;if(pos<=mid)return query(k<<1,pos);else return query(k<<1|1,pos);}int main(){int i,j;while(scanf("%d",&n)==1&&n!=0){build(1,1,n);int a,b;for(i=1;i<=n;i++) { scanf("%d%d",&a,&b); update(1,a,b);} for(i=1;i<n;i++) printf("%d ",query(1,i)); printf("%d\n",query(1,n));}return 0; }

树状数组解法:
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n;
int num[1000005];
int lowbit(int x)
{
return x&(-x);
}
void add(int i ,int val)
{
while(i<=n)
{
num[i]=num[i]+val;
i=i+lowbit(i);
}
}
int sum(int end)
{
int sum=0;
while(end>0)
{
sum=sum+num[end];
end=end-lowbit(end);
}
return sum;
}
int main()
{
while(scanf("%d",&n),n)
{
memset(num,0,sizeof(num));
int a, b;
for(int j=1;j<=n;j++)
{
scanf("%d%d",&a,&b);
add(a,1);
add(b+1,-1);
}
for(int i=1;i<n;i++)
printf("%d ",sum(i));
printf("%d\n",sum(n));
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: