hdu 3874 线段树 单点更新，成段求和

线段树做法，成段求和，单点更新

View Code

#include<cstdio>
#include<cstring>
#include<map>
#include<algorithm>
using namespace std;
const int maxn = 50010;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
__int64 sum[maxn<<2];
struct node{
    int l,r,id;
}p[200010];
map<int,int> hash;
void pushup(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int p,int val,int l,int r,int rt)
{
    if(l==r){
        sum[rt]+=val;
        return ;
    }
    int m=(l+r)>>1;
    if(p<=m) update(p,val,lson);
    else update(p,val,rson);
    pushup(rt);
}
__int64 query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        //printf("rt=%d sum[rt]=%I64d\n",rt,sum[rt]);
        return sum[rt];
    }
    int m=(l+r)>>1;
    __int64 ret=0;
    if(L<=m)  ret+=query(L,R,lson);
    if(R>m)   ret+=query(L,R,rson);
    return ret;
}
int cmp(node a,node b)
{
    return a.r<b.r;
}
int a[maxn];
__int64 ans[200010];
int main()
{
    int t,i,j,l,r,n;
    int q;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++) scanf("%d",&a[i]);
        scanf("%d",&q);
        for(i=1;i<=q;i++)
        {
            scanf("%d%d",&p[i].l,&p[i].r);
            p[i].id=i;
        }
        int pt=1;
        hash.clear();
        sort(p+1,p+q+1,cmp);
        memset(sum,0,sizeof(sum));
        for(i=1;i<=q;i++)
        {
            while(pt<=p[i].r)
            {
                if(hash[a[pt]]!=0)
                {
                    update(hash[a[pt]],-a[pt],1,n,1);
                }
                update(pt,a[pt],1,n,1);
                hash[a[pt]]=pt;
                pt++;
            }
        //    printf("sum=%I64d\n",sum[3]);
            ans[p[i].id]=query(p[i].l,p[i].r,1,n,1);
        }
        for(i=1;i<=q;i++)
        {
            printf("%I64d\n",ans[i]);
        }
    }
    return 0;
}