hdu 3874 树状数组

 http://acm.hdu.edu.cn/showproblem.php?pid=3874

来源：

2011 Multi-University Training Contest 4 - Host by SDU

 

题目：给你一个n（50000）个元素的数组，元素都是非负整数（1000000），然后有q个询问（200000），对每个询问x,y问[x, y]之间的和（注意相同的元素只计算一次）。。。

 

分析：这题比赛期间始终没做出来，想到了对x排序，然后每次往后走的时候更新l和r的值，这样l只是扫描一遍，但是r却依然是n*n。。。果断杯具

其实这题不难，先对y排序，记录每一个值前一个出现的位置，然后每次一个值v时，若前面出现过v则将前面一个v灭掉，然后将当前位置加上v。。。其实就是保留最后一个出现v的位置的v，前面的地方都归0.。。。唉，很简单的树状数组了。。总是想不出来啊，思想总是跟不上。。。。

 

。。。一一对应得hash用map<int, int>来做相当方便。。。。

 

代码：

 

 

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<memory.h>
#include<math.h>
#include<map>
using namespace std;

const int N=50010;
int n, m, a[N];
map<int, int> hash;
__int64 ans[N*4], su[N];
struct node
{
	int i, l, r;
} q[N*4];


void update(int i, int v)
{
	while(i<=n)
	{
		su[i] += v;
		i += i&(-i);
	}
}

__int64 query(int i)
{
	__int64 sum=0;
	while(i>0)
	{
		sum += su[i];
		i -= i&(-i);
	}
	return sum;
}

int cmp1(node &a, node &b)
{
	return a.r<b.r;
}

int main()
{
	int cas, i, j, k, l, r;
	scanf("%d", &cas);
	while(cas--)
	{
		scanf("%d", &n);
		for(i=1; i<=n; i++)
			scanf("%d", &a[i]);
		scanf("%d", &m);
		for(i=0; i<m; i++)
		{
			scanf("%d%d", &q[i].l, &q[i].r);
			if(q[i].l>q[i].r)
				swap(q[i].l, q[i].r);
			q[i].i = i;
		}
		sort(q, q+m, cmp1);

		for(i=0; i<=n; i++)
			su[i] = 0;
		hash.clear();
		r = 1;
		for(i=0; i<m; i++)
		{
			while(r<=q[i].r)
			{
				if(hash[a[r]]!=0)
					update(hash[a[r]], -a[r]);
				hash[a[r]] = r;
				update(r, a[r]);
				r++;
			}
			ans[q[i].i] = query(q[i].r) - query(q[i].l-1);
		}
		
		for(i=0; i<m; i++)
			printf("%I64d\n", ans[i]);
	}
	return 0;
}