[蓝桥杯][2016年第七届真题]压缩变换（主席树求区间不同数的个数）

题目描述
小明最近在研究压缩算法。
他知道，压缩的时候如果能够使得数值很小，就能通过熵编码得到较高的压缩比。
然而，要使数值很小是一个挑战。

最近，小明需要压缩一些正整数的序列，这些序列的特点是，后面出现的数字很大可能是刚出现过不久的数字。对于这种特殊的序列，小明准备对序列做一个变换来减小数字的值。

变换的过程如下：
从左到右枚举序列，每枚举到一个数字，如果这个数字没有出现过，刚将数字变换成它的相反数，如果数字出现过，则看它在原序列中最后的一次出现后面（且在当前数前面）出现了几种数字，用这个种类数替换原来的数字。

比如，序列(a1, a2, a3, a4, a5)=(1, 2, 2, 1, 2)在变换过程为：
a1: 1未出现过，所以a1变为-1；
a2: 2未出现过，所以a2变为-2；
a3: 2出现过，最后一次为原序列的a2，在a2后、a3前有0种数字，所以a3变为0；
a4: 1出现过，最后一次为原序列的a1，在a1后、a4前有1种数字，所以a4变为1；
a5: 2出现过，最后一次为原序列的a3，在a3后、a5前有1种数字，所以a5变为1。
现在，给出原序列，请问，按这种变换规则变换后的序列是什么。
输入
输入第一行包含一个整数n，表示序列的长度。
第二行包含n个正整数，表示输入序列。
输出
输出一行，包含n个数，表示变换后的序列。
样例输入
5
1 2 2 1 2
样例输出
-1 -2 0 1 1
思路：这个题目我不知道有没有别的做法，但是主席树是可以做的。这算是一个求区间不同数的个数的模板题了。注意区间端点的取值。
代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int maxx=1e5+100;
struct node{
	int l,r,sum;
}p[maxx*40];
int root[maxx*40];
int a[maxx];
int n,rot=0;

inline int build(int l,int r)
{
	int cur=++rot;
	p[cur].sum=0;
	p[cur].l=p[cur].r=0;
	if(l==r) return cur;
	int mid=l+r>>1;
	p[cur].l=build(l,mid);
	p[cur].r=build(mid+1,r);
	return cur;
}
inline int update(int pos,int lrot,int v,int l,int r)
{
	int cur=++rot;
	p[cur]=p[lrot];
	p[cur].sum+=v;
	if(l==r) return cur;
	int mid=l+r>>1;
	if(pos<=mid) p[cur].l=update(pos,p[lrot].l,v,l,mid);
	else p[cur].r=update(pos,p[lrot].r,v,mid+1,r);
	return cur;
}
inline int query(int pos,int lrot,int l,int r)
{
	if(l==r) return p[lrot].sum;
	int mid=l+r>>1;
	if(pos<=mid) return p[p[lrot].r].sum+query(pos,p[lrot].l,l,mid);
	else return query(pos,p[lrot].r,mid+1,r);
}
int main()
{
	scanf("%d",&n);
	map<int,int> mp;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	root[0]=build(1,n);
	for(int i=1;i<=n;i++)
	{
		if(mp[a[i]]==0) root[i]=update(i,root[i-1],1,1,n);
		else
		{
			int t=update(mp[a[i]],root[i-1],-1,1,n);
			root[i]=update(i,t,1,1,n);
			
		}
		mp[a[i]]=i;
	}
	mp.clear();
	for(int i=1;i<=n;i++)
	{
		if(mp[a[i]]==0) cout<<-a[i]<<" ";
		else cout<<query(mp[a[i]]+1,root[i-1],1,n)<<" ";
		mp[a[i]]=i;
	}
	return 0;
}

努力加油a啊，(^o)/~