1.题目描述
给你一个有序整数数组,数组中的数可以是正数、负数、零,请实现一个函数,这个函数返回一个整数:返回这个数组所有数的平方值中有多少种不同的取值。举例:
nums = {-1,1,1,1},
那么你应该返回的是:1。因为这个数组所有数的平方取值都是1,只有一种取值
nums = {-1,0,1,2,3}
你应该返回4,因为nums数组所有元素的平方值一共4种取值:1,0,4,9
2.解题思路
使用双指针,从两边向中间扫描。将绝对值大的数字删掉,计数即可,并记录刚才删除的数值的绝对值,以免出现多次相同的数据,重复计数的问题。
3.代码实现
A = [-5,-3,-1,1,1,2]
n = len(A)
i = 0
j = n - 1
pre = float("inf")
count = 0
# 注意这里是小于等于
while i <= j:
if abs(A[i]) > abs(A[j]):
if pre!=abs(A[i]):
count+=1
pre = abs(A[i])
i+=1
else:
if pre!=abs(A[j]):
count+=1
pre = abs(A[j])
j-=1
print (count)