有序數(shù)組的平方

有序數(shù)組的平方

給你一個按 非遞減順序 排序的整數(shù)數(shù)組 nums,返回 每個數(shù)字的平方 組成的新數(shù)組,要求也按 非遞減順序 排序。

示例 1:

輸入:nums = [-4,-1,0,3,10]
輸出:[0,1,9,16,100]
解釋:平方后,數(shù)組變?yōu)?[16,1,0,9,100]
排序后,數(shù)組變?yōu)?[0,1,9,16,100]

示例 2:

輸入:nums = [-7,-3,2,3,11]
輸出:[4,9,9,49,121]

方法一:直接排序

public int[] sortedSquares(int[] nums) {
    int[] res = new int[nums.length];
    for (int i = 0; i < nums.length; i++) {
        res[i] = nums[i] * nums[i];
    }
    Arrays.sort(res);
    return res;
}

方法二:雙指針
根據(jù)有序特點,先找到最后一個負(fù)數(shù)的位置,對于平方,其左邊的結(jié)果從右往左遞增,其右邊的結(jié)果從左往右遞增

public int[] sortedSquares(int[] nums) {
    int[] res = new int[nums.length];
    int lastNegative = 0;
    for (int i = 0; i < nums.length; i++) {
        if (nums[i] < 0) {
            lastNegative = i;
        } else {
            break;
        }
    }
    int left = lastNegative, right = lastNegative + 1;
    int i = 0;
    while (left >= 0 || right < nums.length) {
        int mul1 = left >= 0 ? nums[left] * nums[left] : Integer.MAX_VALUE;
        int mul2 = right < nums.length ? nums[right] * nums[right] : Integer.MAX_VALUE;
        if (mul1 < mul2) {
            res[i++] = mul1;
            left--;
        } else {
            res[i++] = mul2;
            right++;
        }
    }
    return res;
}

雙指針2
數(shù)組平方從兩邊向中間遞減

public int[] sortedSquares(int[] nums) {
    int[] res = new int[nums.length];
    int left = 0, right = nums.length - 1;
    int i = nums.length - 1;
    while (left <= right) {
        int mul1 = nums[left] * nums[left];
        int mul2 = nums[right] * nums[right];
        if (mul1 > mul2) {
            res[i--] = mul1;
            left++;
        } else {
            res[i--] = mul2;
            right--;
        }
    }
    return res;
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容