线段树单点更新-杭电3874

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 200010
struct node
{
	int r,l,pos;
}p[maxn];
long long sum[maxn<<2];
int visit[1000010];
int a[50010];
long long num[maxn];
bool cmp(node x,node y)
{
	return x.r<y.r;
}
void up(int rt)
{
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
long long query(int L,int R,int rt,int a,int b)
{
	if(a<=L&&R<=b)
	return sum[rt];
	int m=(L+R)/2;
	long long ans=0;
	if(a<=m)ans+=query(L,m,rt<<1,a,b);
	if(b>m)ans+=query(m+1,R,rt<<1|1,a,b);
	return ans;
}
void update(int L,int R,int rt,int p,int x)
{
	if(L==R)
	{
		sum[rt]+=x;
		return;
	}
	int m=(L+R)/2;
	if(p<=m)
	update(L,m,rt<<1,p,x);
	else update(m+1,R,rt<<1|1,p,x);
	up(rt);
}

int main()
{
	int N,M,T,i,num1;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&N);
		memset(sum,0,sizeof(sum));
		memset(visit,-1,sizeof(visit));
		for(i=1;i<=N;i++)
		{
			scanf("%d",&a[i]);
		}
		scanf("%d",&M);
		for(i=1;i<=M;i++)
		{
			scanf("%d %d",&p[i].l,&p[i].r);
			p[i].pos=i;
		}
		sort(p+1,p+1+M,cmp);
		num1=1;
		for(i=1;i<=M;i++)
		{
			while(num1<=p[i].r)
			{
			if(visit[a[num1]]!=-1)update(1,N,1,visit[a[num1]],-a[num1]);
			visit[a[num1]]=num1;
			update(1,N,1,num1,a[num1]);
			num1++;
			}
			num[p[i].pos]=query(1,N,1,p[i].l,p[i].r);
		}
		for(i=1;i<=M;i++)
		{
			printf("%I64d\n",num[i]);
		}
	}
return 0;
 }