HDU 1556 Color the ball 很典型的更新区间查找点的题（线段树树状数组两种解法）

这道题应该说是线段树和树状数组简单的应用吧！都是用到了基本的功能更新区间查找点！

当使用线段树时，不想原来更新点查找区间那么简单，这里需要在结构体内另外添加一个lnc用来记录更新区间时正好覆盖到该区域时的改变值！然而在查找区间过程中当所查找的区间并不是给定的区间时，需要把给定的区间的sum值进行更新改变！并且也把该子区间的调用update进行更新，这样才能维护一个更新区间查找区间的线段树！

用树状数组时需要注意此时read（i）就是读取下表为i的值为read（i）！

注意数组尽量开大点！避免出现wa或是re！

下面是ac代码给大家参考：

线段树做法：

#include<iostream>
using namespace std;
struct node
{
int left,right;
int sum,lnc;
}tree[300000];
int num[100000];
void init(int id,int l,int r)
{
if(l==r)
{
tree[id].sum=num[l];
tree[id].left=tree[id].right=l;
tree[id].lnc=0;
return ;
}
init(id*2,l,(l+r)/2);
init(id*2+1,(l+r)/2+1,r);
tree[id].sum=tree[2*id].sum+tree[id*2+1].sum;
tree[id].lnc=0;
tree[id].left=l;
tree[id].right=r;
}
void update(int id,int l,int r,int val)
{
if(tree[id].left==l&&tree[id].right==r)
{
tree[id].lnc+=val;
return ;
}
tree[id].sum+=val*(r-l+1);
int mid=(tree[id].left+tree[id].right)/2;
if(r<=mid)
update(2*id,l,r,val);
else if(l>=mid+1)
update(2*id+1,l,r,val);
else
{
update(2*id,l,mid,val);
update(2*id+1,mid+1,r,val);
}
}
int read(int id,int l,int r)
{
if(tree[id].left==l&&tree[id].right==r)
{
return tree[id].sum+tree[id].lnc*(r-l+1);
}
tree[id].sum+=tree[id].lnc*(tree[id].right-tree[id].left+1);
int val=tree[id].lnc;
int mid=(tree[id].left+tree[id].right)/2;
update(2*id,tree[id].left,mid,val);
update(2*id+1,mid+1,tree[id].right,val);
tree[id].lnc=0;
if(r<=mid)
return read(2*id,l,r);
else if(l>=mid+1)
return read(2*id+1,l,r);
else
{
return read(2*id,l,mid)+read(2*id+1,mid+1,r);
}
}
int main()
{
int n,i,a,b,temp;
while(scanf("%d",&n)&&n)
{
memset(num,0,sizeof(num));
memset(tree,0,sizeof(tree));
init(1,1,n);
temp=n;

while(temp--)
{
scanf("%d%d",&a,&b);
update(1,a,b,1);
}
for(i=1;i<=n;i++)
{
printf("%d",read(1,i,i));
if(i!=n)
printf(" ");
else
printf("/n");
}
}
return 0;
}

 

树状数组做法：

#include<iostream>
using namespace std;
int tree[100100];
int maxn;
void update(int k,int v)
{
while(k<=maxn)
{
tree[k]+=v;
k+=k&-k;
}
}
int read(int k)//求区间1到k的和值
{
int sum=0;
while(k>0)
{
sum+=tree[k];
k-=k&-k;
}
return sum;
}
int main()
{
int i,a,b,temp;
while(scanf("%d",&maxn)&&maxn)
{
temp=maxn;
memset(tree,0,sizeof(tree));
while(temp--)
{
scanf("%d%d",&a,&b);
update(a,1);
update(b+1,-1);
}
for(i=1;i<=maxn;i++)
{
printf("%d",read(i));
if(i!=maxn)
printf(" ");
else
printf("/n");
}
}
return 0;
}

 

哈哈保持激情加油了！