题目链接:HDU 3874 Necklace
【题目大意】
组成项链的珠子有品级之分,价值从1~1000000不等,然而珠子的价值会有重复,例如一个项链有这6个珠子构成 1 1 1 2 3 1,
我们要求给定区间珠子的价值总和,如果区间内有相同价值的珠子,则只计算相同价值珠子中的一个;
用 F(左区间,右区间),来求区间和。
例如 F(1,3)=1, F(2,4)=3, F(2,6)=6.
区间求和问题用树状数组来做比较方便,但是重复数字怎么处理那? 在网上学到了离线操作,非常巧妙
离线查询,顾名思义,不再是实时查询,而是根据需要先调整查询的顺序。
简单来讲就是将查询操作按照右区间从小到大排序。
然后从头依次遍历, 每次用map储存当前数字的最右区间,如果发现当前map值不为空,就删除节点(删除之前保存的数字最右区间),这样在求和的时候就不会重复计算相同的值了。
具体来看代码吧~
【源代码】
#include <iostream>
#include <cstdio>
#include <map>
#include <cstring>
#include <algorithm>
using namespace std;
int n;
typedef long long ll;
const int maxn = 200010;
const int maxe = 50010;
struct node{
int l,r,po;
}order[maxn];
ll s[maxe];
int num[maxe];
ll ans[maxn];
bool cmp(const node&a,const node&b){
return a.r<b.r; //按照右区间从小到大排序
}
int lowbit(int x)
{
return x&(-x);
}
void modify(int x,int val){
for(int i=x;i<=n;i+=lowbit(i)){
s[i]+=val;
}
}
ll sum(int x){
ll ans = 0;
for(int i=x;i>0;i-=lowbit(i))
ans+=s[i];
return ans;
}
int main(){
int T;
map<int ,int >mp;
scanf("%d",&T);
while(T--){
memset(s,0,sizeof(s)); //储存树状数组要清空
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&num[i]);
}
int cmd;
scanf("%d",&cmd);
for(int i=1;i<=cmd;i++){
scanf("%d%d",&order[i].l,&order[i].r);
order[i].po = i; //记录下表,方便排序后寻找
}
sort(order+1,order+cmd+1,cmp);
int r=1; //从开始依次遍历
mp.clear();
for(int i=1;i<=cmd;i++){
while(r<=order[i].r){
if(mp[num[r]])
modify(mp[num[r]],-num[r]); //如果出现重复, 删去之前的节点,map中保存了上次出现的节点位置
mp[num[r]]=r; //保存当前节点位置
modify(r,num[r]); //更新树状数组
r++;
}
ans [order[i].po]=sum(order[i].r)-sum(order[i].l-1); //每次直接求和
}
for(int i=1;i<=cmd;i++)
printf("%lld\n",ans[i]);
}
return 0;
}