有序数组平方后有多少个不同的数

package Array;

public class SquareNotEqual{
    /**
     * 一个有序数组，平方之后，有多少个不一样的数
     * 思路:先把他们全部转换为非负数，然后两个指针从两端开始向中间扫描，从大到小开始计数，此时头尾两指针的元素得比较大小，并且记录
     * 上次的数据，防止重复计数。
     */
     public static int squareNotEqual(int[] arr){
         if(arr == null || arr.length == 0){
             return 0;
         }
         if(arr.length == 1){
             return 1;
         }
         for(int i = 0; i < arr.length; i++){//将所有负数转正
             arr[i] =  Math.abs(arr[i]);
         }
         int head = 0, tail = arr.length - 1;
         int lastBig = -1;//上一个数据
         int count = 0;//不一样的数的个数
         while(head <= tail){//注意这里是小于等于
            while(arr[head] == lastBig){
                head++;
            }
            while(arr[tail] == lastBig){
                tail--;
            }
            if(head <= tail){
                count++;
                if(arr[head] > arr[tail]){
                    lastBig = arr[head];
                    head++;
                }else{
                    lastBig = arr[tail];
                    tail--;
                }
            }else{
                return count;
            }
         }
         return count;
     }
     public static void main(String[] args){
         int[] arr = {-5,-3,-1,1,1,1,2};
         int ans = squareNotEqual(arr);
         System.out.println(ans);
     }
}