LeetCode——905. 按奇偶排序数组

题目描述

给你一个整数数组 nums，将 nums 中的的所有偶数元素移动到数组的前面，后跟所有奇数元素。

返回满足此条件的 任一数组 作为答案。

示例 1：

输入：nums = [3,1,2,4]
输出：[2,4,3,1]
解释：[4,2,3,1]、[2,4,1,3] 和 [4,2,1,3] 也会被视作正确答案。

示例 2：

输入：nums = [0]
输出：[0]

提示：

• 1 <= nums.length <= 5000
• 0 <= nums[i] <= 5000

答案

我的答案

class Solution {
    public int[] sortArrayByParity(int[] nums) {
        int[] arr = new int[nums.length];
        int left = 0;
        int right = nums.length-1;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i]%2==0){
                arr[left++] = nums[i];
            }else {
                arr[right--] = nums[i];
            }
        }
        return arr;
    }
}

官方答案

方法一：两次遍历

思路

新建一个数组 res 用来保存排序完毕的数组。遍历两次 nums，第一次遍历时把所有偶数依次追加到 res 中，第二次遍历时把所有奇数依次追加到 res 中。

代码

class Solution {
    public int[] sortArrayByParity(int[] nums) {
        int n = nums.length, index = 0;
        int[] res = new int[n];
        for (int num : nums) {
            if (num % 2 == 0) {
                res[index++] = num;
            }
        }
        for (int num : nums) {
            if (num % 2 == 1) {
                res[index++] = num;
            }
        }
        return res;
    }
}

复杂度分析

• 时间复杂度：O(n)，其中 n 为数组 nums 的长度。需遍历 nums 两次次。

• 空间复杂度：O(1)。结果不计入空间复杂度。

方法二：双指针 + 一次遍历

思路

记数组 nums 的长度为 n。方法一需要遍历两次 nums，第一次遍历时遇到奇数会跳过，第二次遍历时遇到偶数会跳过，这部分可以优化。

新建一个长度为 n 的数组 res 用来保存排完序的数组。遍历一遍 nums，遇到偶数则从 res 左侧开始替换元素，遇到奇数则从 res 右侧开始替换元素。遍历完成后，res 就保存了排序完毕的数组。

代码

class Solution {
    public int[] sortArrayByParity(int[] nums) {
        int n = nums.length;
        int[] res = new int[n];
        int left = 0, right = n - 1;
        for (int num : nums) {
            if (num % 2 == 0) {
                res[left++] = num;
            } else {
                res[right--] = num;
            }
        }
        return res;
    }
}

复杂度分析

• 时间复杂度：O(n)，其中 nn 为数组 nums 的长度。只需遍历 nums 一次。

• 空间复杂度：O(1)。结果不计入空间复杂度。

方法三：原地交换

思路

记数组 nums 的长度为 n。先从 nums 左侧开始遍历，如果遇到的是偶数，就表示这个元素已经排好序了，继续从左往右遍历，直到遇到一个奇数。然后从 nums 右侧开始遍历，如果遇到的是奇数，就表示这个元素已经排好序了，继续从右往左遍历，直到遇到一个偶数。交换这个奇数和偶数的位置，并且重复两边的遍历，直到在中间相遇，nums 排序完毕。

代码

class Solution {
    public int[] sortArrayByParity(int[] nums) {
        int left = 0, right = nums.length - 1;
        while (left < right) {
            while (left < right && nums[left] % 2 == 0) {
                left++;
            }
            while (left < right && nums[right] % 2 == 1) {
                right--;
            }
            if (left < right) {
                int temp = nums[left];
                nums[left] = nums[right];
                nums[right] = temp;
                left++;
                right--;
            }
        }
        return nums;
    }
}

复杂度分析

• 时间复杂度：O(n)。原数组中每个元素只遍历一次。

• 空间复杂度：O(1)。原地排序，只消耗常数空间。