题目
给定一个非负整数数组 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] 也会被接受。
法一:两次遍历
写法一:设两个可变长度的list,一个存放偶数,一个存放奇数。然后呢,遍历A数组,将偶数下标对应元素设为偶数,奇数下标对应元素设为奇数.
static public int[] sortArrayByParityII(int[] A) {
List<Integer> a = new ArrayList<Integer>();
List<Integer> b = new ArrayList<Integer>();
for (int i = 0; i < A.length; i++) {
if (A[i] % 2 == 0) {
a.add(A[i]);
} else {
b.add(A[i]);
}
}
int index = 0;
for (int i = 0; i < A.length; i++) {
A[i++] = a.get(index);
A[i] = b.get(index);
index++;
}
return A;
}
写法二:通过下标来操作
public int[] sortArrayByParityII2(int[] A) {
int[] ans = new int[A.length];
int index = 0;
for (int i : A) {
if (i % 2 == 0) {
ans[index] = i;
index += 2;
}
}
index = 1;
for (int i : A) {
if (i % 2 != 0) {
ans[index] = i;
index += 2;
}
}
return ans;
}
法二:双指针
设置i从0开始,j从1开始,i+=2,j+=2,i可以遍历该数组的偶数下标,j可以遍历该数组的奇数下标。
因为题目说了“ A 中一半整数是奇数,一半整数是偶数”,所以如果对偶数位排好序,那么奇数位也相应排好了。
// 双指针
public int[] sortArrayByParityII3(int[] A) {
int j = 1;
for (int i = 0; i < A.length; i += 2) {
if (A[i] % 2 != 0) {
while (A[j] % 2 != 0) {
j += 2;
}
swap(A, i, j);
}
}
return A;
}
public void swap(int[] A, int i, int j) {
int temp = A[i];
A[i] = A[j];
A[j] = temp;
}
反思
如何降低时间复杂度,如何让代码更高效,是我刷题必须要考虑的东西。
学习别人的思想,然后吸收,并学以致用。
end.