一、题目描述
小明最近在研究压缩算法。
他知道,压缩的时候如果能够使得数值很小,就能通过熵编码得到较高的压缩比。
然而,要使数值很小是一个挑战。
最近,小明需要压缩一些正整数的序列,这些序列的特点是,后面出现的数字很大可能是刚出现过不久的数字。对于这种特殊的序列,小明准备对序列做一个变换来减小数字的值。
变换的过程如下:
从左到右枚举序列,每枚举到一个数字,如果这个数字没有出现过,刚将数字变换成它的相反数,如果数字出现过,则看它在原序列中最后的一次出现后面(且在当前数前面)出现了几种数字,用这个种类数替换原来的数字。
比如,序列(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
数据规模与约定
对于30%的数据,n<=1000;
对于50%的数据,n<=30000;
对于100%的数据,1 <=n<=100000,1<=ai<=10^9
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 3000ms
二、解题思路
解题思路很简单,但是也出现很多小错误导致花了较多时间。
进行压缩很简单,设置一个列表a,存放我们输入的数字。定义一个列表b,存放结果。对a记性遍历,如果a已经存在b中,调用函数check找到种类数目,将种类数目返回,使用方法append()添加到b中。如果a的数字在b中还没有出现,则将a中的数字乘以(-1)添加到b中即可。
对于得到种类的函数check,我们定义一个数组c进行统计。同时我们需要得到数字在原数列中最后一次出现的位置。可以使用find()函数,但是find函数的使用对象是字符串,而我们已有数列是列表。并且需要注意的是,join()函数不能直接将包含整数的列表转换为字符串。所以对a进行遍历再使用join函数即可。find函数是查找字符串第一次出现的位置,所以可以使用rfind()函数,从右边查找第一次出现的位置就可以得到最后出现的位置。
难点一一攻破,我们来看看代码。
三、代码
n = int(input())
a = list(map(int,input().split()))
b = []
def check(x):
index = 0
num = 0
c = []
index = "".join([str(i) for i in a])[:x].rfind(str(a[x]))
for i in range(index+1,x):
if a[i] not in c:
c.append(a[i])
num += 1
return num
for i in range(n):
if a[i]*(-1) not in b:
b.append(a[i]*(-1))
else:
b.append(check(i))
for i in b:
print(i," ",end='')