树状数组的题,求区间内不同的数的和,离线处理,有个小技巧,按照右边查询r排序,每次都将相同的数更新到最右边,这样每次查询就可以计算不同的数的和。
#include<cstdio>
#include<cstring>
#include<map>
#include<cmath>
#include<algorithm>
#define LL __int64
#pragma comment(linker,"/STACK:1024000000,1024000000")
using namespace std;
const int maxn = 50010;
const int maxx = 200010;
const int maxxx=1000010;
int T,n,m;
LL b[maxn];
int a[maxn],last[maxxx];
LL js[maxx];
struct node
{
int l,r,i;
bool operator <(const node &n)const
{
return r<n.r;
}
}c[maxx];
int lowbit(int x)
{
return x&(-x);
}
LL sum(int x)
{
LL sum=0;
while(x)
{
sum+=b[x];
x-=lowbit(x);
}
return sum;
}
void update(int i,int num)
{
while(i<=n)
{
b[i]+=num;
i+=lowbit(i);
}
}
int main()
{
scanf("%d",&T);
while(T--)
{
for(int i=0; i<maxn; i++)b[i]=0,a[i]=0;
for(int i=0; i<maxx; i++)js[i]=0,c[i].i=0;
for(int i=0; i<maxxx; i++)last[i]=0;
scanf("%d",&n);
for(int i=1; i<=n; i++)
{
scanf("%d",&a[i]);
}
scanf("%d",&m);
for(int i=0; i<m; i++)
{
scanf("%d %d",&c[i].l,&c[i].r);
c[i].i=i;
}
sort(c,c+m);
int lastr=1;
for(int i=0; i<m; i++)
{
for(int j=lastr; j<=c[i].r; j++)
{
if(last[a[j]]==0)
{
update(j,a[j]);
last[a[j]]=j;
}
else if(last[a[j]]!=j)
{
update(j,a[j]);
update(last[a[j]],-a[j]);
last[a[j]]=j;
}
}
lastr=c[i].r;
js[c[i].i]=sum(c[i].r)-sum(c[i].l-1);
}
for(int i=0; i<m; i++)
{
printf("%I64d\n",js[i]);
}
}
return 0;
}