HDU3333(线段树/数状数组+hash)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3333

题意:数轴上的区间和问题,有一些变化就是要求的是不同元素的和,线段树or数状数组可以解决.

刚开始时竟然给线段树的每人结点维护了一个set 来标记是否有重复,那个复杂度弄的比暴力都高了吧,汗一个~

后来看了下别人的思路,采用离线操作,对Q所要求的区间以右端点递增排序,从左至右插入新的元素,如果该元素已经插入了线段树中,就删掉旧位置的,在新位置插入该元素,并对以该元素位置为右端点的区间进行查询求和,即为结果.

记录元素是否已插入线段树可用离散化+二分来实现,也可用hash来记录.这里用的是hash表.

首先用线段树写了,可是一直RE的很悲状,后来才发现ans[]数组竟然开小了,这鬼~

数状数组实现很简单,甚至效率也高了线段树很多.

线段树:

#include<stdio.h>#include<string.h>#include<stdlib.h>#define M 30007#define N 100048#define L(x) (x<<1)#define R(x) ((x<<1)|1)#define Mid(x,y) ((x+y)>>1)typedef struct{int lt,rt;__int64 sum;}SegT;typedef struct talHash{int ix,v;talHash *next;}HashN;typedef struct{int r,l,ix;}Pair;SegT tr[M*4];HashN *hash[M];HashN MEM[M];Pair pair[N];int a[M],e;__int64 ans[N],anst;int Cmp(const void *a,const void *b){return *(int *)a-*(int *)b;}int Key(int value){value+=value;while(value>=M) value-=M;return value;}int AtHash(int value){HashN *cur;for(cur=hash[Key(value)];cur;cur=cur->next)if(cur->v==value) break;return cur?cur->ix:0;}void HashInsert(int value,int ix){HashN *cur=&MEM[e++];cur->ix=ix;cur->v=value;cur->next=hash[Key(value)];hash[Key(value)]=cur;}void Create(int ix,int l,int r){tr[ix].lt=l;tr[ix].rt=r;tr[ix].sum=0;if(l==r) return ;int mid=Mid(l,r);Create(L(ix),l,mid);Create(R(ix),mid+1,r);}void Updata(int ix,int pos,int delta){int mid;if(tr[ix].lt==tr[ix].rt){tr[ix].sum+=delta;return ;}mid=Mid(tr[ix].lt,tr[ix].rt);if(pos<=mid)Updata(L(ix),pos,delta);else Updata(R(ix),pos,delta);tr[ix].sum=tr[L(ix)].sum+tr[R(ix)].sum;}void Query(int ix,int l,int r){int mid;if(tr[ix].lt==l&&tr[ix].rt==r){ anst+=tr[ix].sum;return ;}mid=Mid(tr[ix].lt,tr[ix].rt);if(r<=mid) Query(L(ix),l,r);else if(l>mid) Query(R(ix),l,r);else{Query(L(ix),l,mid);Query(R(ix),mid+1,r);}}int main(){int n,t,i,j,q;int preix;scanf("%d",&t);while(t--){memset(hash,NULL,sizeof(hash));scanf("%d",&n);for(i=1;i<=n;i++)scanf("%d",a+i);scanf("%d",&q);for(i=0;i<q;i++){scanf("%d%d",&pair[i].l,&pair[i].r);pair[i].ix=i;}qsort(pair,q,sizeof(pair[0]),Cmp);Create(1,1,n);for(i=1,j=e=0;i<=n;i++){preix=AtHash(a[i]);if(preix)Updata(1,preix,-a[i]);HashInsert(a[i],i);Updata(1,i,a[i]);for(;j<q&&pair[j].r==i;j++){anst=0;Query(1,pair[j].l,pair[j].r);ans[pair[j].ix]=anst;}}for(j=0;j<q;j++)printf("%I64d/n",ans[j]);}return 0;} 

数状数组:

#include<stdio.h>#include<string.h>#include<stdlib.h>#define M 30007#define N 100048#define LowBit(x) (x&(-x))typedef struct talHash{int ix,v;talHash *next;}HashN;typedef struct{int r,l,ix;}Pair;HashN *hash[M];HashN MEM[M];Pair pair[N];__int64 C[M];int a[M],e;__int64 ans[N],anst;int Cmp(const void *a,const void *b){return *(int *)a-*(int *)b;}int Key(int value){value+=value;while(value>=M) value-=M;return value;}int AtHash(int value){HashN *cur;for(cur=hash[Key(value)];cur;cur=cur->next)if(cur->v==value) break;return cur?cur->ix:0;}void HashInsert(int value,int ix){HashN *cur=&MEM[e++];cur->ix=ix;cur->v=value;cur->next=hash[Key(value)];hash[Key(value)]=cur;}void Insert(int ix,int delta){for(;ix<M;ix+=LowBit(ix)) C[ix]+=delta;}__int64 Sum(int ix){__int64 ret=0;for(;ix>0;ix-=LowBit(ix)) ret+=C[ix];return ret;}int main(){int n,t,i,j,q;int preix;scanf("%d",&t);while(t--){memset(hash,NULL,sizeof(hash));scanf("%d",&n);for(i=1;i<=n;i++)scanf("%d",a+i);scanf("%d",&q);for(i=0;i<q;i++){scanf("%d%d",&pair[i].l,&pair[i].r);pair[i].ix=i;}qsort(pair,q,sizeof(pair[0]),Cmp);memset(C,0,sizeof(C));for(i=1,j=e=0;i<=n;i++){preix=AtHash(a[i]);if(preix)Insert(preix,-a[i]);HashInsert(a[i],i);Insert(i,a[i]);for(;j<q&&pair[j].r==i;j++)ans[pair[j].ix]=Sum(pair[j].r)-Sum(pair[j].l-1);}for(j=0;j<q;j++)printf("%I64d/n",ans[j]);}return 0;}