HDU 3333 Turing Tree --树状数组+离线处理

时间:2023-12-25 22:48:07

题意:统计一段序列【L,R】的和,重复元素只算一次。

解法:容易看出在线做很难处理重复的情况,干脆全部讲查询读进来,然后将查询根据右端点排个序,然后离散化数据以后就可以操作了。

每次读入一个数,如果这个数之前出现过,那么删除之前出现的那个数,改加上这个数,然后进行所有右端点小于等于此时下标的查询即可。

关于正确性,引用sdj222555的话来说,"观察一个区间,我们可以发现,如果出现重复的,尽量删除左边的,保留右边的,那么右端点相同的区间都可以进行查询。"

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <map>
#define lll __int64
using namespace std;
#define N 30007 lll c[N],ans[*N+];
map<int,int> mp;
int a[N],pos[N],n,b[N],d[N];
struct Query
{
int L,R,ind;
}Q[*N+]; int cmp(Query ka,Query kb)
{
return ka.R < kb.R;
} inline int lowbit(int x) { return x&(-x); } void modify(int pos,lll val)
{
while(pos <= n)
{
c[pos] += val;
pos += lowbit(pos);
}
} lll getsum(int pos)
{
lll ans = ;
while(pos > )
{
ans += c[pos];
pos -= lowbit(pos);
}
return ans;
} int main()
{
int t,i,j,q;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(i=;i<=n;i++)
scanf("%d",&a[i]),b[i] = a[i];
scanf("%d",&q);
for(i=;i<=q;i++)
scanf("%d%d",&Q[i].L,&Q[i].R),Q[i].ind = i;
memset(c,,sizeof(c));
sort(b+,b+n+);
sort(Q+,Q+q+,cmp);
int ind = unique(b+,b+n+)-b-;
for(i=;i<=ind;i++)
mp[b[i]] = i;
for(i=;i<=n;i++) //d[i]为a[i]离散化后的数
d[i] = mp[a[i]];
memset(pos,,sizeof(pos)); //pos 存上次出现的位置
j = ;
for(i=;i<=n;i++)
{
if(pos[d[i]]) modify(pos[d[i]],-a[i]);
modify(i,a[i]);
pos[d[i]] = i;
while(j <= q && Q[j].R == i)
{
ans[Q[j].ind] = getsum(Q[j].R)-getsum(Q[j].L-);
j++;
}
if(j > q)
break;
}
for(i=;i<=q;i++)
printf("%I64d\n",ans[i]);
}
return ;
}