leetcode75——Sort Colors

题目大意:将给出的数组升序排序(数组中只有0、1、2三种数字)

分析:排序应用。大多数排序算法也只能达到O(nlogn)的复杂度,而我们想在O(n)的复杂度内完成排序,就需要另一种方法:双指针。这道题可以使用双指针就特殊在他只有三种数字进行排序。我们设定两个指针zero和two分别指向数组最左和最右,他们分别维持着0和2的区域(也就是下一个要被放0/2的区域)。遍历数组时遇到0就放到数组左侧,遇到2就放去数组右侧,这样最后1就会都被挤在中间了。

代码:

class Solution {
public:
void sortColors(vector<int>& nums) {
int zero = 0, two = nums.size() - 1;  //下一个应当被放置zero的位置(two)
int i = 0;
while (i <= two) {  //i遍历到two之前就已经排序结束
if (nums[i] == 0) {  //遇到0就交换到zero区域(i++即可,因为交换过来的数字只能是1,1无需操作)
swap(nums[i++], nums[zero++]);
}
else if (nums[i] == 2) {  //遇到2就交换到two区域(i不能动,因为不知道交换过来的数字是什么,可能是0)
swap(nums[i], nums[two--]);
}
else i++;
}
}
};

猜你喜欢

转载自blog.csdn.net/tzyshiwolaogongya/article/details/80363579