hdu 3874 Necklace 线段树单点更新区间求和

http://acm.hdu.edu.cn/showproblem.php?pid=3874

题意：

给你n个数，然后给出m个的询问[l,r]问[l,r]中不同的数的和为多少？

思路：
单纯只考虑不同数的和的话，可能该数在之前出现过，后边又出现了，但是所求的区间不包括前边那个数，那么这个数就要被加进来。关键是如何维护该区间不包含重复计算同一个数。

首先我们离线处理所有询问，然后按r排序，然后遍历一边处理那些还没被加进线段树的数，如果他之前出现过了，那么就将之前的删除，然后再吧后边的加进来，这样既能保证同一个数只被计算的一次了，而且不影响后边的询问。

这里关键是按r排序之后的处理：

#include <iostream>

#include <cstdio>

#include <cmath>

#include <vector>

#include <cstring>

#include <algorithm>

#include <string>

#include <set>

#include <functional>

#include <numeric>

#include <sstream>

#include <stack>

#include <map>

#include <queue>



#define CL(arr, val) memset(arr, val, sizeof(arr))



#define lc l,m,rt<<1

#define rc m + 1,r,rt<<1|1

#define pi acos(-1.0)

#define L(x)    (x) << 1

#define R(x)    (x) << 1 | 1

#define MID(l, r)   (l + r) >> 1

#define Min(x, y)   (x) < (y) ? (x) : (y)

#define Max(x, y)   (x) < (y) ? (y) : (x)

#define E(x)        (1 << (x))

#define iabs(x)     (x) < 0 ? -(x) : (x)

#define OUT(x)  printf("%I64d\n", x)

#define lowbit(x)   (x)&(-x)

#define Read()  freopen("din.txt", "r", stdin)

#define Write() freopen("dout.txt", "w", stdout)





#define M 200007

#define N 50007

#define ll __int64

using namespace std;

const double eps = 1e-10;



struct node

{

    int l,r;

    int id;

}nd[M];



ll val[4*N];

ll ans[M];

int a[N];

int off[1000007];



int n;



void pushup(int rt)

{

    val[rt] = val[rt<<1] + val[rt<<1|1];

}

void build(int l,int r,int rt)

{

    val[rt] = 0;

    if (l == r) return ;

    int m = (l + r)>>1;

    build(lc);

    build(rc);

}

void update(int pos,ll sc,int l,int r,int rt)

{

    if (l == r)

    {

        val[rt] += sc;

        return ;

    }

    int m = (l + r)>>1;

    if (pos <= m) update(pos,sc,lc);

    else update(pos,sc,rc);

    pushup(rt);

}

ll query(int L,int R,int l,int r,int rt)

{

    if (l >= L && r <= R) return val[rt];

    ll res = 0;

    int m = (l + r)>>1;

    if (L <= m) res += query(L,R,lc);

    if (R > m) res += query(L,R,rc);

    return res;

}



int cmp(node a,node b) { return a.r < b.r; }

int cmp2(node a,node b) { return a.id < b.id; }



int main()

{

//    Read();

    int i;

    int T,n,m;

    scanf("%d",&T);

    while (T--)

    {

        scanf("%d",&n);

        for (i = 1; i <= n; ++i) scanf("%d",&a[i]);

        scanf("%d",&m);

        for (i = 0; i < m; ++i)

        {

            scanf("%d%d",&nd[i].l,&nd[i].r);

            nd[i].id = i;

        }



        sort(nd,nd + m,cmp);

        build(1,n,1);

        CL(off,0);



        int s = 1;

        for (i = 0; i < m; ++i)

        {

            while (s <= nd[i].r)

            {

                if (off[a[s]]) update(off[a[s]],-a[s],1,n,1);



                update(s,a[s],1,n,1);

                off[a[s]] = s;

                s++;

            }

            ans[nd[i].id] = query(nd[i].l,nd[i].r,1,n,1);

        }



        for (i = 0; i < m; ++i)

        {

            printf("%I64d\n",ans[i]);

        }

    }

    return 0;

}

View Code