LeetCode 922. 按奇偶排序数组 II（C、C++、python）

---

给定一个非负整数数组 A， A 中一半整数是奇数，一半整数是偶数。

对数组进行排序，以便当 A[i] 为奇数时，i 也是奇数；当 A[i] 为偶数时， i 也是偶数。

你可以返回任何满足上述条件的数组作为答案。

 

示例：

输入：[4,2,5,7]
输出：[4,5,2,7]
解释：[4,7,2,5]，[2,5,4,7]，[2,7,4,5] 也会被接受。

 

提示：

2 <= A.length <= 20000

A.length % 2 == 0

0 <= A[i] <= 1000

C

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* sortArrayByParityII(int* A, int ASize, int* returnSize) 
{
    int n=ASize;
    int* M=(int*)malloc(sizeof(int)*n/2);
    int* N=(int*)malloc(sizeof(int)*n/2);
    int* res=(int*)malloc(sizeof(int)*n);
    int k=0;
    int r=0;
    for(int i=0;i<n;i++)
    {
        if(A[i]%2==1)
        {
            M[k]=A[i];
            k++;
        }
        else
        {
            N[r]=A[i];
            r++;
        }        
    }
    for(int i=0;i<n;i++)
    {
        if(i%2==0)
        {
            res[i]=N[i/2];
        }
        else
        {
            res[i]=M[i/2];
        }
    }
    *returnSize=n;
    return res;
}

C++

class Solution {
public:
    vector<int> sortArrayByParityII(vector<int>& A) 
    {
        int n=A.size();
        vector<int> M;
        vector<int> N;
        for(int i=0;i<n;i++)
        {
            if(A[i]%2==1)
            {
                M.push_back(A[i]);
            }
            else
            {
                N.push_back(A[i]);
            }
        }
        vector<int> res;
        for(int i=0;i<n;i++)
        {
            if(i%2==0)
            {
                res.push_back(N[i/2]);
            }
            else
            {
                res.push_back(M[i/2]);
            }
        }
        return res;
    }
};

python

class Solution:
    def sortArrayByParityII(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        n=len(A)
        M=[]
        N=[]
        for i in range(n):
            if A[i]%2==1:
                M.append(A[i])
            else:
                N.append(A[i])
        res=[]
        for i in range(n):
            if i%2==0:
                res.append(N[i//2])
            else:
                res.append(M[i//2])
        return res