HDU 3874 Necklace 树状数组的应用

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

题意：有一些数，这些数中有重复的，问从[L,R]区间的和是多少，重复的数只能算一次。

思路：因为有多次询问，所以暴力的话肯定超时，又因为是区间求和问题，所以可以考虑用树状数组求。树状数组可以解决没有重复数的情况，因此这道题我们可以特殊处理一下。首先我们可以把所有的询问都存起来，然后对询问按右端点排序，然后每次询问，对于排序后的每次询问，我们只考虑到该区间的右端点，并且记录每个数最后出现的位置，若出现相同的数，则需要从该数的位置开始减去该数。

代码：

#include <iostream>
#include <cstdio>
#include <string.h>
#include <algorithm>
#include <map>
#include <vector>
#include <string>
#include <set>
using namespace std;
const int N = 50010;
const int M = 200010;
int dit
;
__int64 num
;
__int64 ans[M];
struct interval{
    int lp,rp,id;
}rr[M];
bool  cmp(interval a,interval b){
    if(a.rp == b.rp)
        return a.lp < b.lp;
    return a.rp < b.rp;
}
int inline lowbit(int x){
    return x & (-x);
}
void inline update(int x,int add){
    while(x < N){
      num[x] += add;
      x += lowbit(x);
    }
}
__int64 inline sum(int x){
    __int64 s = 0;
    while(x > 0){
      s += num[x];
      x -= lowbit(x);
    }
    return s;
}
int main(){
    int numcase;
    map<int,int> mp;
    scanf("%d",&numcase);
    while(numcase--){
       int n,m;
       scanf("%d",&n);
       memset(dit,0,sizeof(dit));
       for(int i = 1; i <= n; ++i){
          scanf("%d",&dit[i]);
       }
       scanf("%d",&m);
       for(int i = 0; i < m; ++i){
          scanf("%d%d",&rr[i].lp,&rr[i].rp);
          if(rr[i].rp < rr[i].lp)
             swap(rr[i].lp,rr[i].rp);
          rr[i].id = i;
       }
       sort(rr,rr+m,cmp);
       memset(num,0,sizeof(num));
       memset(ans,0,sizeof(ans));
       mp.clear();
       int rp = 1;
       for(int i = 0; i < m; ++i){
           while(rp <= rr[i].rp){
              int x = dit[rp];
              if(mp[x] != 0){
                 update(mp[x],-x);
              }
              update(rp,x);
              mp[x] = rp;
              rp++;
           }
           ans[rr[i].id] = sum(rr[i].rp) - sum(rr[i].lp - 1); 
       }
       for(int i = 0; i < m; ++i){
          printf("%I64d\n",ans[i]);
       }
       
    }
    //system("pause");
    return 0;
}