小明最近在研究压缩算法。
他知道,压缩的时候如果能够使得数值很小,就能通过熵编码得到较高的压缩比。
然而,要使数值很小是一个挑战。
最近,小明需要压缩一些正整数的序列,这些序列的特点是,后面出现的数字很大可能是刚出现过不久的数字。对于这种特殊的序列,小明准备对序列做一个变换来减小数字的值。
变换的过程如下:
从左到右枚举序列,每枚举到一个数字,如果这个数字没有出现过,刚将数字变换成它的相反数,如果数字出现过,则看它在原序列中最后的一次出现后面(且在当前数前面)出现了几种数字,用这个种类数替换原来的数字。
比如,序列(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
再例如,输入:
12
1 1 2 3 2 3 1 2 2 2 3 1
程序应该输出:
-1 0 -2 -3 1 1 2 2 0 0 2 2
本题其实题意非常简单,主要难点在于如何处理大数据的情况,本题暴力解法,就是map记住上次的位置,遍历中间,统计不同,大概可以过50%的数据
本题正解是利用线段树来解题,本题正满足线段树求和的要求
本题突破点,本题修改是从后向前,是从最后一层向上修改
注意维护更新,本题还是不是很好想到线段树的,具体解释附在代码上
本代码参考:压缩变换 蓝桥杯 java_xiao_you_you的博客-CSDN博客
代码如下
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StreamTokenizer;
import java.util.HashMap;
import java.util.Map;
public class Main压缩变换 {
static int teg[],n,beg;
public static void main(String[] args) throws IOException {
StreamTokenizer x = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
x.nextToken();
n=(int)x.nval;
int aa[]=new int[n];
teg=new int[4*n];
Map<Integer, Integer> p=new HashMap<Integer, Integer>();//用来记录上次这个数的位置
beg=1;
while(beg<n) beg<<=1;
for(int i=0;i<n;i++) {
x.nextToken();
int k=(int)x.nval;
if(!p.containsKey(k)) {//如果没出现过
aa[i]=-k;
p.put(k, i);
}
else {//出现过
aa[i]=query(0, beg, p.get(k)+1, i, 0);//从根节点开始向下累加
update(p.get(k),-1);//包括这个节点的集合,内容全部-1,消除重复
/*
* 解释,因为本题题意就是上一个和本数一样的的节点,那么现在因为出现了下一个跟他相同的节点
* 那么本来这个数的贡献值就被归为现在新的位置相同的数了
* 所以原本这个数的贡献值要删除避免重复累加
* 1 2 5 3 2 1
* 后面节点是一定会包括这个新节点的
* 因为是一直一直向后拓展,那么是只有两种情况,要么没贡献,要么就包括这个位置
* 这也是本题可用线段树解决的原因
* */
p.put(k, i);//更新位置
}
update(i,1);//包括这个点的集合全部+1,更新贡献
}
for(int i=0;i<n;i++) {
if(i!=n-1)
out.print(aa[i]+" ");
else
out.println(aa[i]);
}
out.flush();
}
public static void update(int k,int d) {
k+=beg-1;
teg[k]+=d;
while(k>0) {
k=(k-1)>>1;//向上寻找父节点
teg[k]+=d;
}
}
public static int query(int pl,int pr,int l,int r,int p) {//pl pr是集合范围,l r是我们要查的范围
if(l==r||pr<=l||pl>=r) return 0;
if(l<=pl&&r>=pl) return teg[p];
int sum=0;
int mod=(pl+pr)>>1;
sum+=query(pl,mod,l,r,(p<<1)+1);//向左
sum+=query(mod,pr,l,r,(p+1)<<1);//向右
return sum;
}
}