1.题目描述
给定一个有序数组,请算出平方后的数组有多少个不相同的数字。
例子,[-1,3,3],返回结果 2.
例子,[-1,-1,1,1],返回结果 1.
2.思路
使用双指针,对left和right的绝对值进行比较,如果不相等,那么绝对值大的一边进行收缩,并且不相等的数加一,但是要记录一下这个数的绝对值,这样碰到相同的可以越过去。比如[-1,3,3],一开始right绝对值为3,left绝对值为1,不相等,res为1,记录pre为3,右边向里面收缩一位,然后right绝对值还是3,left绝对值为1,发现right和pre相等,越过,右边向里面收缩一位,right绝对值为1,left绝对值为1,res为2,循环结束。
3.代码
int findDiff(vector<int> nums) {
int res = 0;
int left = 0;
int right = nums.size() - 1;
int pre = -1;
while(left < right) {
if(abs(nums[left]) > abs(nums[right])) {
if(pre != abs(nums[left]) {
res++;
pre = abs(nums[left];
}
left++;
}
else {
if(pre != abs(nums[right]) {
res++;
pre = abs(nums[right];
}
right--;
}
}
return res;
}
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(1)