算法4—双指针法解决有序数组平方和的排序

示例

给个有序数组(包括负数) int[]{-5,1,3,4} 输出结果是 [1, 9, 16, 25]

思路分析

这里采用双指针来实现上述功能，两个指针分别指向数组的最前，最后元素，遍历这个数组，然后两个指针同时往中间移动，取出指向的元素计算平方和，比较两个值谁的大，把大的放入到新数组的最后一个位置(按照自然顺序排序)，然后重复前面步骤即可。示意图如下所示：

1、两个指针的初始位置如下所示，指向数组的最前面，和最后面，同时初始化一个新数组和长度和原数组相等

2、接着比较 head、和 tail 指向的元素平方和谁的大，明显是 25 大，所以把 25 放入到新数组的最后位置，然后 head 往右边移动，指向下一个元素 1，tail 不变，如下图示：

3、此时继续比较 head 指向的和 tail 指向的谁的平方和大，就把谁的值放入到新数组中，显然是 16，所以把 16 放入到新数组中，此时 tail 往左边移动，指向了下一个元素 3，head 不变，如下图示：

4、继续比较 head 和 tail 指向的元素平方和谁的大，此时显然是 9 的大，所以把 9 放入到新数组中，然后 tail
往左移动，指向下一个元素 1，此时和 head 指向了同一个元素，如下图示：

5、最后因为 head 和 tail 都不能再移动了，然后判断 1 和 1 就是相等的，所以直接丢进新数组的 0 号位置即可，然后跳出循环，得到了一个自然顺序的新数组，如下图示：

代码实现

package main.suanfa;

import java.util.Arrays;

public class PingFangDemo {

    /**
     * 计算有序数组平方之后然后升序排序
     * 思路一：先计算平房、然后采用快速排序
     * 思路二：双指针思路
     * 1、定义头尾指针,分别指向数组第一个元素和最后一个元素
     * 2、定义一个新数组用来接受元素
     * 3、比较头尾指针指向的元素平方和谁的大，就把该元素放到数组的最后一个位置，因为是倒叙排序的所以把最大的直接就放到了新数组的最后位置
     * 4、然后移动头、或者尾指针继续重复步骤3
     */
    public static void main(String[] args) {
        int[] arr = new int[]{-5,1,3,4};
        int[] sort = new int[arr.length];
        int k = arr.length-1;
        for (int i = 0,j=arr.length-1; i <= j;) {
            // 比较头尾指针指向的元素平方和谁的大，就移动对应的指针
            if (arr[i] * arr[i] > arr[j] * arr[j]) {
                sort[k--] = arr[i]*arr[i];
                i++;
            }
            sort[k--] = arr[j]*arr[j];
            j--;
        }
        System.out.println("Arrays.toString(sort) = " + Arrays.toString(sort));
    }
}

输出结果如下：

Arrays.toString(sort) = [1, 9, 16, 25]

Process finished with exit code 0