HDU 3333&&HDU-3874 Necklace 离线树状数组

#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;
int N,M;
int num[50050];
long long C[50050];
struct Node{
int L;
int R;
int id;
}P[200020];
long long ans[200020];
int lowbit(int x){
return x&(-x);
}
long long sum(int x){
long long ret=0;
while (x>0) {
ret+=C[x];x-=lowbit(x);
}
return ret;
}
int add(int x,long long d){
while (x<=N){
C[x]+=d;x+=lowbit(x);
}
return 0;
}
long long query(int L,int R){
return sum(R)-sum(L-1);
}
int cmp(Node a,Node b){
return a.R<b.R;
}
int main (){
int T;scanf("%d",&T);
while (T--){
map <int ,int > vis;
vis.clear();
scanf("%d",&N);
for (int i=0;i<N;i++)
scanf("%d",&num[i]),C[i]=0;
C
=0;
scanf("%d",&M);
for (int i=0;i<M;i++){
scanf("%d%d",&P[i].L,&P[i].R),P[i].id=i;
if (P[i].L>P[i].R){
int c=P[i].R;
P[i].R=P[i].L;
P[i].L=c;
}
}
sort(P,P+M,cmp);
int j=0;
for (int i=0;i<M;i++){
while (j<N&&j+1<=P[i].R){
if (vis[num[j]]!=0) {
add(vis[num[j]],-num[j]);
vis[num[j]]=j+1;
add(vis[num[j]],num[j]);
}
else {
vis[num[j]]=j+1;
add(vis[num[j]],num[j]);
}
j++;
}
ans[P[i].id]=query(P[i].L,P[i].R);
}
for (int i=0;i<M;i++){
printf("%I64d\n",ans[i]);
}
}
return 0;
}