SPOJ-DQUERY 题意是求区间内不同数的个数
HDU 3333是求区间内不同数的和
做法基本上差不多,
先存起区间来,然后按区间右端点由小到大排序
然后遍历每个区间
先依次把序列插入到线段树中(到区间右端点停止),在这个过程中,用book数组记录这个数之前出没出现过 出现过的话book数组记录的是该数的位置,没出现过的话就是0,(HDU这道题数的范围是1e9因此需要离散化,SPOJ不用),如果出现过,就把原来位置更新成0并把现在的位置更新,没出现过只需要更新现在的位置,book数组也要更新。插入到右端点停止之后查询一下记录下来就可以了
SPOJ
#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e5+5;
struct node{
int l,r;
int val;
}tree[maxn<<2];
void pushup(int m)
{
tree[m].val=tree[m<<1].val+tree[m<<1|1].val;
}
/*void pushdown(int m)
{
if(tree[m].lazy)
{
tree[m<<1].lazy+=tree[m].lazy;
tree[m<<1|1].lazy+=tree[m].lazy;
tree[m<<1].val+=(tree[m<<1].r-tree[m<<1].l+1)*tree[m].lazy;
tree[m<<1|1].val+=(tree[m<<1|1].r-tree[m<<1|1].l+1)*tree[m].lazy;
tree[m].lazy=0;
}
}*/
void build(int m,int l,int r)
{
tree[m].l=l;
tree[m].r=r;
if(l==r)
{
tree[m].val=0;
return ;
}
int mid=(l+r)>>1;
build(m<<1,l,mid);
build(m<<1|1,mid+1,r);
pushup(m);
}
void update(int m,int index,int val)
{
if(tree[m].l==index&&tree[m].r==index)
{
tree[m].val=val;
return;
}
int mid=(tree[m].l+tree[m].r)>>1;
if(index<=mid)
update(m<<1,index,val);
else
update(m<<1|1,index,val);
pushup(m);
}
int query(int m,int l,int r)
{
if(tree[m].l==l&&tree[m].r==r)
{
return tree[m].val;
}
int mid=(tree[m].l+tree[m].r)>>1;
if(r<=mid)
return query(m<<1,l,r);
else if(l>mid)
return query(m<<1|1,l,r);
else
return query(m<<1,l,mid)+query(m<<1|1,mid+1,r);
}
struct qujian{
int l,r,id;
}qj[200005];
bool cmp(qujian a,qujian b)
{
return a.r<b.r;
}
int n,m,a[maxn],book[1000005],ans[200005];
int main()
{
scanf("%d",&n);
build(1,1,n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d",&qj[i].l,&qj[i].r);
qj[i].id=i;
}
sort(qj,qj+m,cmp);
int pos=1;
for(int i=0;i<m;i++)
{
while(pos<=n&&pos<=qj[i].r)
{
if(book[a[pos]])
update(1,book[a[pos]],0);
update(1,pos,1);
book[a[pos]]=pos;
pos++;
}
ans[qj[i].id]=query(1,qj[i].l,qj[i].r);
}
for(int i=0;i<m;i++)
printf("%d\n",ans[i]);
return 0;
}
HDU
#include<iostream>
#include<cstdio>
#include<vector>
#include<string.h>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=1e5+5;
struct node{
int l,r;
ll value;
}tree[maxn<<2];
void pushup(int m)
{
tree[m].value=tree[m<<1].value+tree[m<<1|1].value;
}
void build(int m,int l,int r)
{
tree[m].l=l;
tree[m].r=r;
tree[m].value=0;
if(l==r)
return;
int mid=(l+r)>>1;
build(m<<1,l,mid);
build(m<<1|1,mid+1,r);
pushup(m);
}
void updata(int m,int index,ll val)
{
if(tree[m].l==index&&tree[m].r==index)
{
tree[m].value=val;
return;
}
int mid=(tree[m].l+tree[m].r)>>1;
if(index<=mid) updata(m<<1,index,val);
else updata(m<<1|1,index,val);
pushup(m);
}
ll query(int m,int l,int r)
{
if(tree[m].l==l&&tree[m].r==r)
return tree[m].value;
int mid=(tree[m].l+tree[m].r)>>1;
ll temp;
if(r<=mid)temp=query(m<<1,l,r);
else if(l>mid) temp=query(m<<1|1,l,r);
else temp=query(m<<1,l,mid)+query(m<<1|1,mid+1,r);
return temp;
}
struct QUJIAN{
int l,r,id;
}qujian[maxn];
bool cmp2(QUJIAN x,QUJIAN y)
{
return x.r<y.r;
}
ll ans[maxn],temp[maxn],a[maxn],n,book[maxn];
int getid(int x)
{
return lower_bound(temp+1,temp+n+1,x)-temp-1;
}
int main()
{
int T,m;
scanf("%d",&T);
while(T--)
{
scanf("%lld",&n);
build(1,1,n);
memset(book,0,sizeof book);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
temp[i]=a[i];
}
sort(temp+1,temp+n+1);
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d",&qujian[i].l,&qujian[i].r);
qujian[i].id=i;
}
sort(qujian,qujian+m,cmp2);
int k=1;
for(int i=0;i<m;i++)
{
while(k<=n&&k<=qujian[i].r)
{
int t=getid(a[k]);
if(book[t])
updata(1,book[t],0);
updata(1,k,a[k]);
book[t]=k;
k++;
}
ans[qujian[i].id]=query(1,qujian[i].l,qujian[i].r);
}
for(int i=0;i<m;i++)
printf("%lld\n",ans[i]);
}
}
本文详细介绍了如何使用线段树解决区间内不同数的个数和和的问题,通过将区间按右端点排序,遍历每个区间,利用book数组记录数的出现情况,实现SPOJ-DQUERY和HDU3333题目的高效解答。

480

被折叠的 条评论
为什么被折叠?



