LeetCode922 按奇偶排序树组II(排序)

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/No__names/article/details/94595480

题目描述如下:

给定一个非负整数数组 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

思路如下:

按照题目描述,由于数组元素中奇数偶数各占一半,问题等价于:奇数下标的数组元素为奇数,偶数下标的数组元素为偶数。

于是遍历偶数下标的数组元素,若发现其值为奇数(不合要求),则开始遍历奇数下标的数组元素,寻找其中值为偶数的元素(也不合要求),将这两个元素的值交换,然后继续对偶数元素的遍历(下一次需要遍历奇数下标元素时,注意从上次进行过交换的奇数下标开始进行遍历,因为前面遍历过的部分一定无误,避免重复,这也是代码中index的作用)。时间复杂度O(n),空间复杂度O(n)。

代码如下:

class Solution {
public:
    vector<int> sortArrayByParityII(vector<int>& A) {
        int index = 1;
	int tmp;
	for (int i = 0; i < A.size(); i += 2) {
		if (A[i] % 2 != 0) {
			for (int j = index; j < A.size(); j += 2) {
				if (A[j] % 2 == 0) {
					tmp = A[i];
					A[i] = A[j];
					A[j] = tmp;
					index = j+2;
					break;
				}
			}
		}
	}
	return A;
    }
};

猜你喜欢

转载自blog.csdn.net/No__names/article/details/94595480
今日推荐