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

点我看题

题意:给你一个长度为n的序列,m次询问,问从l到r不重复数值的和为多少.

分析:用树状数组求和,离线操作去重.先说一下离线操作吧(今天才看的离线,不喜勿喷...),离线操作就是在已知所有输入的情况下,先求出每次的结果,再一次性进行输出.这个题离线操作的方法就是先把每个数第一次出现的位置记录下来,然后把要查询的区间按照从r由小到大排序,排好序之后,在上一个查询区间的右端点到本次查询的右端点之间,不断的更新num的位置,将位置置为num在当前区间最后一次出现的位置,并且撤销前面num的后续影响,这样就能保证查询的时候所有的数往前只出现一次了.任何进行树状数组求和操作就ok了~

参考代码:

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>

using namespace std;
typedef long long ll;
const int maxn = 5e4+10;//数的个数最大值
const int maxm = 2e5+10;//查询次数的最大值
const int maxv = 1e6+10;//数值的最大值
int n,m;//数的个数和查询次数
int val[maxn];//数值
ll arr[maxn];//树状数组中的真正数组,因为它为value的多项之和,要设为long long型,不然可能溢出
ll ans[maxm];//最终结果,也要定义为long long,并且最大长度为maxm
int pos[maxv];//每个数值当前最后一次出现的位置,最大长度为数值的最大值
struct Node{
	int l,r;//查询的左右端点
	int index;//记录第多少次查询
};
Node q[maxm];//数组长度为查询的最大次数

bool cmp( Node a, Node b)
{
	if( a.r != b.r)
		return a.r < b.r;
	return a.l < b.l;
}

inline int lowbit( int x)
{
	return x&-x;
}

void Add( int x, int num)
{
	while( x <= n)
	{
		arr[x] += num;
		x += lowbit(x);
	}
}

ll Sum( int x)//返回值要为long long类型
{
	ll ans = 0;
	while( x)
	{
		ans += arr[x];
		x -= lowbit(x);
	}
	return ans;
}

int main()
{
	int T;
	scanf("%d",&T);
	while( T--)
	{
		memset(arr,0,sizeof(arr));
		memset(pos,0,sizeof(pos));
		scanf("%d",&n);
		for( int i = 1; i <= n; i++)
		{
			scanf("%d",&val[i]);
			Add(i,val[i]);//建立真正的树状数组
			if( !pos[val[i]])//记录当前值第一次出现的位置
				pos[val[i]] = i;
		}

		scanf("%d",&m);
		for( int i = 1; i <= m; i++)
		{
			scanf("%d%d",&q[i].l,&q[i].r);
			q[i].index = i;
		}
		sort(q+1,q+1+m,cmp);//按照查询右端点的值从小到大排序
		
		int R = 1;
		for( int i = 1; i <= m; i++)
		{
			for( int j = R; j <= q[i].r; j++)
			{
				if( pos[val[j]] != j)//表明val[j]不是当前查询最后一次出现的,就说明前面还存在同样的值,那么就要把前面相同的值删掉,并且更新val[i]的位置
				{
					Add(pos[val[j]],-val[j]);//取消前面重复的值
					pos[val[j]] = j;//更新位置
				}
			}
			R = q[i].r;
			ans[q[i].index] = Sum(q[i].r)-Sum(q[i].l-1);//计算结果
		}
		
		for( int i =1; i <= m; i++)
			printf("%lld\n",ans[i]);
	}

	return 0;
}