问题描述:
给定一个数组arr和一个数num,把数组中小于num的数放在数组的左边,等于num的数放在中间,大于num的数放在右边。
解决思路:
定义两个指针,一个表示左边区域边界,一个表示右边区域边界,遍历数组,若i位置的数小于num,就将这个数与左边界的下一个数交换,左边界加1,i加1,如果大于num,就将这个数与右边界的前一个数交换,右边界减1,i不变(因为此时从右边交换过来的数成为新的i,继续判断)。
Java代码:
public class left_mid_right {
//把小于num的数放在左边,等于num的数放在中间,大于num的数放在右边
public static void zhengli(int[] arr, int num){
int left = 0;
int right = 0;
for(int i = 0; i < arr.length - right; i++){
if(arr[i] < num){ //小于num的数放在左边
swap(arr, left, i);
left++;
}else if(arr[i] > num){//大于num就放到右边,保持i不变,因为此时右边换过来的数还未经过判断
swap(arr, arr.length - 1 - right, i);
right++;
i--;
}
}
}
//交换数组中的两个元素
public static void swap(int[] arr, int a, int b){
int temp = arr[a];
arr[a] = arr[b];
arr[b] = temp;
}
//主方法
public static void main(String[] args) {
//int[] arr = {4, 1, 2, 8, 7, 3, 5, 1, 3, 9, 8, 3, 2};
int[] arr = {4};
int num = 3;
zhengli(arr, num);
System.out.println(Arrays.toString(arr));
}
}
以上内容为个人学习总结,如有错误,欢迎指正。