Given an array A of non-negative integers, half of the integers in A are odd, and half of the integers are even.
Sort the array so that whenever A[i] is odd, i is odd; and whenever A[i] is even, i is even.
You may return any answer array that satisfies this condition.
Example 1:
Input: [4,2,5,7]
Output: [4,5,2,7]
Explanation: [4,7,2,5], [2,5,4,7], [2,7,4,5] would also have been accepted.
原文鏈接：https://leetcode.com/problems/sort-array-by-parity-ii/
給定一個數組，其中奇數和偶數各一半，要求返回的數組中，奇數下標中的元素為奇數，偶數索引下標中的元素為偶數。
最簡單的做法就是將原數組中的元素進行分開存放，然后再將這些元素按照要求組裝起來。
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A) {
int i,even_index=0,odd_index=0;
vector<int> odds,evens;
for(int i : A) {
if(i%2 == 0)
evens.push_back(i);
else
odds.push_back(i);
}
for(i=0;i<A.size();i++){
if(i%2==0) {
A[i] = evens[even_index];
even_index++;
} else {
A[i] = odds[odd_index];
odd_index++;
}
}
return A;
}
};
顯然上述做法并不明智，既浪費了時間又浪費了空間。我們可以在此基礎上優化一下，可以先申請一個與原來數組大小相同的結果數組。遍歷原先的數組，如果遇到了奇數元素就將該元素放置到結果數組中的奇數位置上，如果是偶數元素則放置到結果數組中的偶數位置上。
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A) {
int i,even_index=0,odd_index=1;
vector<int> result(A.size(),0);
for(i=0;i<A.size();i++) {
if(A[i]%2==0) {//如果當前數字為偶數，則將該數字放到結果數組中的偶數索引上
result[even_index] = A[i];
even_index += 2;
} else {
result[odd_index] = A[i];
odd_index += 2;
}
}
return result;
}
};
可以看到上述兩個方法都是O(n)的空間復雜度，那么接下來的優化可以從縮小空間復雜度入手，一個不難想到的做法就是如果當前索引是奇數但是元素不是奇數，那么就找到下一個奇數元素進行交換，偶數也是如此。
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A) {
int i,even_index,odd_index;
for(i=0;i<A.size();i++) {
even_index = odd_index = i;
if(i%2 == 0) {//當前為偶數索引
while(A[even_index]%2!=0 && even_index<A.size()) {//找到下一個偶數
even_index++;
}
swap(A[i],A[even_index]);
} else {//當前為奇數索引
while(A[odd_index]%2==0 && odd_index<A.size()) {//找到下一個奇數
odd_index++;
}
swap(A[i],A[odd_index]);
}
}
return A;
}
};
顯然上面一個方法，冗余了很多循環。我們可以對奇數索引進行檢查，如果是奇數元素的話，就檢查下一個奇數索引，直到發現奇數索引上不是奇數元素停止，然后按照此方法檢查偶數索引，最后將其進行交換。
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A) {
int even_index=0,odd_index=1;
while(even_index<A.size() && odd_index<A.size()) {
while(A[even_index]%2==0&&even_index<A.size()) even_index+=2;//偶數索引不為偶數時停下來
while(A[odd_index]%2==1&&odd_index<A.size()) odd_index+=2;//奇數索引不為奇數時停下來
//cout << A[even_index] << ",index:" << even_index << endl;
//cout << A[odd_index] << ",index:" << odd_index << endl;
if(even_index>A.size() || odd_index>A.size()) break;
swap(A[even_index],A[odd_index]);
}
return A;
}
};
另一種寫法：
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A)
{
for(int i=0,j=1;i<A.size()&&j<A.size();)
{
if(A[i]%2==0)
i+=2;
else if(A[j]%2==1)
j+=2;
else
swap(A[i],A[j]);
}
return A;
}
};
