hdu3874/hdu3333 线段树区间求和

很基础的线段树,求和的时候去掉重复的,用离线处理就ok了。


ACcode:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
using std::sort;

const int nsize=53333;
const int msize=221111;

LL ans[msize];
LL sum[nsize<<2];
int pos[nsize];
int change[nsize];

struct Number
{
    int k;
    int idx;
} num[nsize],tmp[nsize];

struct question
{
    int l,r;
    int idx;
} qu[msize];

bool cmp1(question a1,question a2)
{
    if (a1.r!=a2.r) return a1.r<a2.r;
    return a1.l<a2.l;
}

bool cmp2(Number a1,Number a2)
{
    return a1.k<a2.k;
}

void build(int rt,int l,int r)
{
    sum[rt]=0;
    if (l==r) return ;
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
}

void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void update(int rt,int l,int r,int p,int v)
{
    if (l==r)
    {
        sum[rt]=v;
        return ;
    }
    int m=(l+r)>>1;
    if (p<=m) update(rt<<1,l,m,p,v);
    else update(rt<<1|1,m+1,r,p,v);
    PushUp(rt);
}

LL query(int rt,int l,int r,int L,int R)
{
    if (L<=l&&r<=R) return sum[rt];
    int m=(l+r)>>1;
    LL cnt=0;
    if (L<=m) cnt+=query(rt<<1,l,m,L,R);
    if (R>m)  cnt+=query(rt<<1|1,m+1,r,L,R);
    return cnt;
}

int main()
{
    int T,i,j,x,n,q;
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        for (i=1; i<=n; i++)
        {
            num[i].idx=tmp[i].idx=i;
            scanf("%d",&num[i].k);
            tmp[i].k=num[i].k;
        }
        scanf("%d",&q);
        for (i=1; i<=q; i++)
        {
            qu[i].idx=i;
            scanf("%d %d",&qu[i].l,&qu[i].r);
        }
        build(1,1,n);
        sort(qu+1,qu+q+1,cmp1);
        sort(tmp+1,tmp+n+1,cmp2);
        memset(pos,0,sizeof(pos));
        change[tmp[1].idx]=1;
        for (x=1,i=2; i<=n; i++)
        {
            if (tmp[i].k!=tmp[i-1].k) x++;
            change[tmp[i].idx]=x;
        }
        for (j=i=1; i<=q;)
        {
            while (j<=qu[i].r&&j<=n)
            {
                if (pos[change[num[j].idx]]>0)
                    update(1,1,n,pos[change[num[j].idx]],0);
                pos[change[num[j].idx]]=j;
                update(1,1,n,j,num[j].k);
                j++;
            }
            while (qu[i].r<j&&i<=q)
            {
                ans[qu[i].idx]=query(1,1,n,qu[i].l,qu[i].r);
                i++;
            }
        }
        for (i=1; i<=q; i++)
            printf("%I64d\n",ans[i]);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值