Hdu 3333 （线段树）

来源：互联网发布：远程服务器监控软件编辑：程序博客网时间：2024/05/06 19:57

题目

从别人那学得一种线段树做法，查询某一区间不同数字的和

先按查询的右坐标升序排列。把每个数放进线段树的时候，先判断再之前他有没有在线段树，如果在，则删除它，并把他的位置更新到当前点

#include<stdio.h>#include<string.h>#include<algorithm>#include<map>using namespace std;typedef __int64 ll;#define N 300010#define M 100010struct {    int l,r;    ll num;}root[N*4];struct Q{    int s,t,ind;}q[M];bool cmp(Q i,Q j){    return i.t<j.t;}inline void build(int t,int x,int y){    root[t].l=x;    root[t].r=y;    root[t].num=0;    if(x==y) return;    int m=(x+y)>>1;    build(t*2,x,m);    build(t*2+1,m+1,y);}inline void Modefiy(int t,int x,ll val){    int l=root[t].l;    int r=root[t].r;    if(l==r)    {        root[t].num+=val;return;    }    int m=(l+r)>>1;    if(x<=m) Modefiy(t*2,x,val);    else Modefiy(t*2+1,x,val);    root[t].num=root[t*2].num+root[t*2+1].num;}inline ll query(int t,int x,int y){    int l=root[t].l;    int r=root[t].r;    if(l==x&&r==y)    {        return root[t].num;    }    int m=(l+r)>>1;    ll ans=0;    if(x<=m) ans+=query(t*2,x,min(m,y));    if(y>m) ans+=query(t*2+1,max(m+1,x),y);    return ans;}int t,n,qn;ll a[N];map<ll,int>m;ll ans[M];int main(){    scanf("%d",&t);    while(t--)    {        m.clear();        scanf("%d",&n);        for(int i=1;i<=n;i++) scanf("%I64d",&a[i]);        build(1,1,n);        scanf("%d",&qn);        for(int i=0;i<qn;i++)        {            scanf("%d%d",&q[i].s,&q[i].t);            q[i].ind=i;        }        sort(q,q+qn,cmp);        int k=1;        for(int i=0;i<qn;i++)        {            for(;k<=q[i].t;k++)            {                if(m[a[k]]!=0) Modefiy(1,m[a[k]],-a[k]);                m[a[k]]=k;                Modefiy(1,k,a[k]);            }            ans[q[i].ind]=query(1,q[i].s,q[i].t);        }        for(int i=0;i<qn;i++) printf("%I64d\n",ans[i]);    }    return 0;}