今天看了左神的荷兰国旗问题,照葫芦画瓢写出的
荷兰国旗问题:给定一个数组,和一个数,把小于的放在数的左边,大于的放在右边,等于的放在中间
public static int[] partition(int[] arr,int l,int r,int num){
//小于num的区域
int less = l-1;
//大于num的区域
int more = r+1;
//确保数组不为空或者长度为1
if(l<r){
//结束条件为没有没被查询过的数
while(l<more){
//如果当前查询的数小于给定的num,将小于num的区域的下一位与当前数交换,并且将小于num的区域+1
if(arr[l]<num){
swap(arr,++less,l++);
}else if(arr[l]>num){//同上
swap(arr,--more,l);
}else{//若是遇到等于num的数,直接跳过
l++;
}
}
}
//返回的是等于num的数的区域的下标,也就是从less的下一个,到more的前一个
return new int[]{ less+1,more-1 };
}
//交换数组内两个数的位置
public static void swap(int[] arr,int i,int j){
int temp = 0;
temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}