题目
经典的荷兰国旗问题。
整个数组的范围为0~n-1。现在可以理解为小于等于num的范围边界为从左到-1这个位置。从边界往外逐一比较,如果这个数小于num,则将这个数与边界外的第一个数进行交换,并且将范围向外扩充一个。如果这个数大于num,则不做任何操作,接着往下比较。
在数组的左右两端画两个区域less和more。从左开始遍历数组,如果等于num,则cur++;如果小于num,则与less外的第一个元素交换,并且less++,cur++;如果大于num,则与more外面的第一个元素交换,并且more–,然而cur不动,继续考察。
class Solution {
public void sortColors(int[] nums) {
int l = -1;
int r = nums.length;
int cur = 0;
while(cur<r){
if(nums[cur]==0){
swap(nums,++l,cur++);
}else if(nums[cur]==2){
swap(nums,--r,cur);
}else{
cur++;
}
}
}
public void swap(int[] num,int i,int j){
int tmp = num[i];
num[i] = num[j];
num[j] = tmp;
}
}
要注意和左边界换的时候,换过来的元素一定是等于target的,所以需要cur++,与右边界换的时候,换过来的元素需要继续观察比较,因此要cur按兵不动。