您的位置:首页 > 其它

HDU 3874 Necklace(树状数组的离线操作)

2015-08-25 10:44 435 查看
题目链接: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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: