package question;

import java.util.Arrays;

/*
    @author: Roderland
    @create: 2020-09-23---20:28
*/
public class Question977 {
    public int[] sortedSquares(int[] A) {
        for (int i = 0; i < A.length; i++) {
            A[i] = A[i] * A[i];
        }
        Arrays.sort(A);
        return A;
    }

    public static void main(String[] args) {
        //[-4,-1,0,3,10]
        int[] a = new int[]{-2, 0};
        int[] ints = new Question977().sortedSquares2(a);
        for (int i : ints) {
            System.out.println(i);
        }
    }

    public int[] sortedSquares2(int[] A) {
        if (A.length == 1) return new int[]{A[0] * A[0]};
        int[] a = new int[A.length];
        int i = 0;
        while (A[i] < 0) i++;
        for (int j = 0; j < A.length; j++) A[j] = A[j] * A[j];
        int left = i - 1;
        int right = i;
        int j = 0;
        while (left >= 0 && right < A.length) {
            if (A[left] < A[right]) a[j++] = A[left--];
            else a[j++] = A[right++];
        }
        while (left >= 0) a[j++] = A[left--];
        while (right < A.length) a[j++] = A[right++];
        return a;
    }
}