题源:https://vjudge.net/problem/15630/origin
题意:给你一段长度为n的区间,每个点都有一个值,请你算出某个区间内,不同值之和。
思路:
0、先说这题,看了无数题解终于比着敲出来了,但是就是一直WA,一直WA。最后发现各个参数应该用long long 而不是int 我哭。
1、首先是一些小的分析:我们注意到每个数可以很大(1e9),但是数的数量很小(3e4)。所以我们想到离散化;其次是询问的个数很多(1e5),所以一个个处理可能会T,所以我们想到离线处理,把所有询问都存下来,然后按照右端点大小排序。对于每个问题,每次更新到相应的右端点即可。(或者对于每个右端点,更新到对应的问题即可。)
2、query函数询问的是当前区间不同值之和。
3、pre代表的是,某个值上一次出现的位置是哪里。(因为已经离散化过了,所以值和位置一一对应)。
下面是AC代码。
其中核心代码fun有两种书写方式。都可以用。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#define maxn 30100
#define lowbit(x) (x&(-x))
#define LL long long
using namespace std;
LL n,m;
LL a[maxn],c[maxn],ans[100100];
map <LL,int> pre;
struct node {
LL l,r,id;
} q[100100];
bool cmp(node a,node b) {
return a.r<b.r;
}
void init() {
memset(c,0,sizeof(c));
pre.clear();
}
void updata(LL i,LL k) {
while(i<=n) {
c[i]+=k;
i+=lowbit(i);
}
}
LL query(LL i) {
LL sum=0;
while(i>0) {
sum+=c[i];
i-=lowbit(i);
}
return sum;
}/*
void fun() {
for(int pos=1,i=1; pos<=n; pos++) {
if(pre[lsh[pos]]!=0)
updata(pre[lsh[pos]],-a[pos]);
updata(pos,a[pos]);
pre[lsh[pos]]=pos;
while(q[i].r==pos) {
ans[q[i].id]=query(q[i].r)-query(q[i].l-1);
i++;
}
if(i>m) break;
}
}*/
void fun() { //换一下顺序 以问题为主
//这个pos是遍历的点的编号
for(LL i=1,pos=0; i<=m; i++) { //对于每个问题
while(pos<q[i].r) {
pos++;
if(pre[a[pos]]!=0) { //若之前来过这个点
updata(pre[a[pos]],-a[pos]);
}
updata(pos,a[pos]);
pre[a[pos]]=pos;
}
ans[q[i].id]=query(q[i].r)-query(q[i].l-1);
}
}
int main() {
int T;
cin>>T;
for(int k=1; k<=T; k++) {
init();
cin>>n;
for(LL i=1; i<=n; i++) {
scanf("%lld",&a[i]);
}
cin>>m;
for(LL i=1; i<=m; i++) {
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q+1,q+m+1,cmp);
fun();
for(LL i=1; i<=m; i++) {
printf("%lld\n",ans[i]);
}
}
return 0;
}