学到了一个很好的思路,建立输入顺序和输入值的相对大小的对应关系,就是下面代码中的ref数组
还有就是不要拘泥于模板,这道题的线段树处理只需要一个ll数组就行
两篇不错的博客:
https://blog.csdn.net/dacc123/article/details/52797310
https://www.cnblogs.com/111qqz/p/4712392.html
#include <stdlib.h>
#include <cstdio>
#include <iostream>
#include <queue>
#include <stack>
#include <string.h>
#include <math.h>
#include <algorithm>
#define mem(t, v) memset ((t) , v, sizeof(t))
#define INF 0x3f3f3f3f
#define eps 1e-8
typedef long long ll;
using namespace std;
const int MAX=30010;
int t,n,m;
struct s{
int id;
ll val;
}in[MAX];
int f[MAX]; //数值和新次序建立对应关系
int ref[MAX];
int last[MAX]; //新次序的mark数组
struct z{
int x,y,id;
ll ans;
}q[100010];
ll tree[MAX<<2];
bool cmp(s a,s b){
return a.val<b.val;
}
bool _cmp(s a,s b){
return a.id<b.id;
}
bool cmpq(z a,z b){
return a.y<b.y;
}
bool _cmpq(z a,z b){
return a.id<b.id;
}
void Update(int i,int l,int r,int val,ll tag){
if(l==r){
tree[i]=tag;
return ;
}
int mid=(l+r)>>1;
if(mid>=val)Update(i<<1,l,mid,val,tag);
else Update(i<<1|1,mid+1,r,val,tag);
tree[i]=tree[i<<1]+tree[i<<1|1];
}
ll Search(int i,int l,int r,int L,int R){
if(l>=L&&r<=R)
return tree[i];
if(r<L||l>R)return 0;
ll s=0;
int mid=(l+r)>>1;
if(mid>=L)s+=Search(i<<1,l,mid,L,R);
if(mid+1<=R)s+=Search(i<<1|1,mid+1,r,L,R);
return s;
}
int main(){
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%lld",&in[i].val);
in[i].id=i;
}
sort(in+1,in+n+1,cmp);
int top=0;
in[0].val=-1;
for(int i=1;i<=n;i++){
if(in[i].val!=in[i-1].val){
f[++top]=in[i].val;
}
ref[in[i].id]=top;
}
sort(in+1,in+n+1,_cmp);
// for(int i=1;i<=n;i++)
// printf("%d++\n",ref[i]);
scanf("%d",&m);
for(int i=0;i<m;i++){
scanf("%d%d",&q[i].x,&q[i].y);
q[i].id=i;
}
sort(q,q+m,cmpq);
mem(last,0);
mem(tree,0);
for(int i=0,j=1;i<m;i++){
for(;j<=q[i].y;j++){
int t=ref[in[j].id];
//printf("%d %d %d %d %d %d--\n",j,t,last[t],q[i].x,q[i].y,q[i].id);
if(last[t]){
Update(1,1,n,last[t],0);
}
last[t]=j;
Update(1,1,n,j,in[j].val);
}
q[i].ans=Search(1,1,n,q[i].x,q[i].y);
//printf("%lld**\n",q[i].ans);
}
sort(q,q+m,_cmpq);
for(int i=0;i<m;i++)
printf("%lld\n",q[i].ans);
}
return 0;
}