[主席树]HDOJ3874 Necklace

题意：n个数 m个询问

询问的是[l, r]区间内不同的数的和

没有修改，静态的主席树即可

与 SPOJ QUERY 一样 将重复的元素建树即可

注意范围：$N \le  50000$ 每个值不超过1000000

也就是加起来会爆int 要用LL

  1 #include <bits/stdc++.h>
  2 using namespace std;
  3 typedef long long LL;
  4 #define lson l, m
  5 #define rson m+1, r
  6 const int N=5e4+5;
  7
  8 int L[N<<5], R[N<<5];
  9 LL sum[N<<5];
 10 int tot;
 11 int a[N], T[N];
 12 int read()
 13 {
 14     char ch=‘ ‘;
 15     int ans=0;
 16     while(ch<‘0‘ || ch>‘9‘)
 17         ch=getchar();
 18     while(ch<=‘9‘ && ch>=‘0‘)
 19     {
 20         ans=ans*10+ch-‘0‘;
 21         ch=getchar();
 22     }
 23     return ans;
 24 }
 25
 26 int build(int l, int r)
 27 {
 28     int rt=(++tot);
 29     sum[rt]=0;
 30     if(l<r)
 31     {
 32         int m=(l+r)>>1;
 33         L[rt]=build(lson);
 34         R[rt]=build(rson);
 35     }
 36     return rt;
 37 }
 38
 39 int update(int pre, int l, int r, int x, int val)
 40 {
 41     int rt=(++tot);
 42     L[rt]=L[pre], R[rt]=R[pre], sum[rt]=sum[pre]+val;
 43     if(l<r)
 44     {
 45         int m=(l+r)>>1;
 46         if(x<=m)
 47             L[rt]=update(L[pre], lson, x, val);
 48         else
 49             R[rt]=update(R[pre], rson, x, val);
 50     }
 51     return rt;
 52 }
 53
 54 LL query(int u, int v, int l, int r, int k)
 55 {
 56      if(l>=k)
 57         return sum[v]-sum[u];
 58     int m=(l+r)>>1;
 59     LL ans=0;
 60     if(m>=k)
 61         ans+=query(L[u], L[v], lson, k);
 62     ans+=query(R[u], R[v], rson, k);
 63     return ans;
 64 }
 65
 66 LL all[N];
 67 int main()
 68 {
 69     int t;
 70     scanf("%d", &t);
 71     while(t--)
 72     {
 73         tot=0;
 74         int n=read();
 75 //        scanf("%d", &n);
 76         all[0]=0;
 77         for(int i=1; i<=n; i++)
 78         {
 79 //            scanf("%d", &a[i]);
 80             a[i]=read();
 81             all[i]=all[i-1]+a[i];
 82         }
 83            T[0]=0;
 84         map<int, int> mp;
 85         mp.clear();
 86         for(int i=1; i<=n; i++)
 87         {
 88             if(mp.find(a[i])!=mp.end())
 89                 T[i]=update(T[i-1], 1, n, mp[a[i]], a[i]);
 90             else
 91                 T[i]=T[i-1];
 92             mp[a[i]]=i;
 93         }
 94         int m=read();
 95 //        scanf("%d", &m);
 96         while(m--)
 97         {
 98             int l, r;
 99             l=read(), r=read();
100 //            scanf("%d%d", &l, &r);
101             printf("%I64d\n", all[r]-all[l-1]-query(T[l-1], T[r], 1, n, l));
102         }
103     }
104     return 0;
105 }

HDOJ 3874