题目
给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库的sort函数的情况下解决这个问题。
示例 1：
输入：nums = [2,0,2,1,1,0] 输出：[0,0,1,1,2,2]
示例 2：
输入：nums = [2,0,1] 输出：[0,1,2]
解题
方法一：快排
class Solution { public: void quickSort(vector<int>& nums,int l,int r){ if(l>=r) return; int i=l,j=r; while(i<j){ while(i<j&&nums[j]>=nums[l]) j--; while(i<j&&nums[i]<=nums[l]) i++; swap(nums[i],nums[j]); } swap(nums[i],nums[l]); quickSort(nums,l,i-1); quickSort(nums,i+1,r); } void sortColors(vector<int>& nums) { quickSort(nums,0,nums.size()-1); } };
方法二：将0和2分别移动到两侧
class Solution { public: void sortColors(vector<int>& nums) { int j=0; for(int i=0;i<nums.size();i++){ if(nums[i]==0){ swap(nums[i],nums[j]); j++; } } j=nums.size()-1; for(int i=nums.size()-1;i>=0;i--){ if(nums[i]==2){ swap(nums[i],nums[j]); j--; } } } };
