0 引言
这道题要求在原地数组上进行操作,也就意味着不能新开一个数组,并且仅用常数的情况下去实现效果。因此我们需要在遍历数组的同时就要操作数组,从而实现排序的效果,此题的难点在于如何在遍历的同时进行排序。
1 方法
这段代码通过一次遍历实现对数组中 0、1、2 的排序,核心思想是用两个指针 j 和 k 分别记录当前 1 和 2 的起始位置(或边界),在遍历过程中根据当前元素与前一个元素是否变化,动态调整元素位置:遇到 0、1、2 时分别插入到对应区域,并通过覆盖和前移的方式维持数组中 0 在前、1 在中、2 在后的顺序,从而实现类似"三色排序"的原地排序效果。
cpp
class Solution{
public:
void sortColors(vector<int>& nums) {
int i=0,j=-1,k=-1;
if (nums.size()>1){
for(int i=0;i<nums.size();i++){
if (i==0){
if (nums[i]==1){j=0;continue;}
else if(nums[i]==2){k=0;continue;}
else {continue;}
}
if (nums[i] != nums[i-1]){
if (nums[i] == 1){
if (j==-1 && k!=-1){
nums[i] = nums[i-1];
nums[k] = 1;
j = k;
k++;
}
else if(j==-1 && k==-1) j = i;
else{
nums[i] = nums[i-1];
nums[k] = 1;
k++;
}
}
else if (nums[i]==2){
if (k==-1) k = i;
}
else{
if (j==-1 && k==-1)continue;
else if(j==-1&&k!=-1){
nums[i] = nums[i-1];
nums[k] = 0;
k++;
}
else if(j!=-1 && k!=-1){
nums[i] = nums[i-1];
nums[j] = 0;
nums[k] = 1;
j++,k++;
}
else if (j!=-1 && k==-1){
nums[i] = nums[i-1];
nums[j] = 0;
j++;
}
}
}
}
}
}
};