题目如下:
给你一个有序整数数组,数组中的数可以是正数、负数、零,请实现一个函数,这个函数返回一个整数:返回这个数组所有数的平方值中有多少种不同的取值。举例:
nums = {-1,1,1,1},
那么你应该返回的是:1。因为这个数组所有数的平方取值都是1,只有一种取值
nums = {-1,0,1,2,3}
你应该返回4,因为nums数组所有元素的平方值一共4种取值:1,0,4,9
那如何利用好有序这个条件呢?
解思路法如下:
绝对值相等可能有哪些情况呢?
两个同符号数的绝对值相等,这也意味着在有序数组中这个两个数是相邻的,这时只需要移动指针跳过相邻相等的数即可。
两个异符号数的绝对值相等,所以我们需要维护两个指针,一个指针从前往后移动,一个从后往前移。原因是前面的负数绝对值可能与后面整数的绝对值相等,我们需要比较前后两个指针绝对值的大小。
综上,我们需要维护两个指针,i开始指向数组第一个元素,i=0;j开始指向数组最后一个元素,j = nums.length-1。i,j指针的另外含义是:数组中索引小于i和大于j的元素都已经被处理了;i,j指向的是未处理元素中绝对值最大的两个元素。当i>j的时候表明所有的元素都已经被处理了,循环结束。
那么如何移动指针呢?
如果nums[i]与nums[j]的绝对值相等,此时执行i++,直到nums[i]的绝对值不等于nums[j]的绝对值(跳过相邻重复元素);j也是做类似的移动,只不过j是向前移,j--;最后计数器加1。
如果nums[i]的绝对值大于nums[j]的绝对值,我们移动指针i,并且计数器加1。原因?因为i和j指向未处理元素中绝对值最大的那个,nums[i]的绝对值已经是未处理元素中绝对值最大的两个数,也就是不可能存在一个未被处理的元素,它的绝对值与nums[i]的绝对值相等。所以这种情况下我们移动i指针。注意这里所说的移动指的是:执行i++直到nums[i]的绝对值发生了改变(跳过相邻相等的元素)。
如果nums[j]的绝对值大,那么与上面类似,移动j指针。
具体代码实现如下:
public int handle (int[] nums) {
if(nums==null || nums.length==0)
return 0;
// result的缩写,最后的返回值
int res = 0;
//i是前指针;j是后指针
int i = 0;
int j = nums.length - 1;
while (i <= j) {
int num1=Math.abs(nums[i]);
int num2=Math.abs(nums[j]);
if (num1 > num2) {//移动i
// 这两个数的绝对值不相等
res += 1;
while(i<=j && Math.abs(nums[i])==num1)
//过滤掉相邻的绝对值相等的数
i++;
} else if (num1 < num2) {
// 这两个数的绝对值不相等
res += 1;
while(i<=j && Math.abs(nums[j])==num2)
//过滤掉相邻的绝对值相等的数
j--;
} else {
res += 1;
while(i<=j && Math.abs(nums[i])==num1)//去重
i++;
while(i<=j && Math.abs(nums[j])==num2)//去重
j--;
}
}
return res;
}