荷兰国旗问题
问题一
给定一个数组arr，和一个数num，请把小于等于num的数放在数 组的左边，大于num的 数放在数组的右边。要求额外空间复杂度O(1)，时间复杂度O(N)
问题二
给定一个数组arr，和一个数num，请把小于num的数放在数组的 左边，等于num的数放 在数组的中间，大于num的数放在数组的 右边。要求额外空间复杂度O(1)，时间复杂度 O(N)
思路
- 定义三个指针（int）
- 一个指针为遍历指针，遍历整个数组
- 一个指针确定左边更少的边界
- 一个指针确定右边最小的边界
注意事项
- 第一个要注意的事情是在传入边界的时候，要将左边界减一，右边界加一。因为在之后交换的时候，是会先将这两个指针都进行变化。
int less = l - 1;
int more = r + 1;
- 第二要注意的是返回两个下标，可以将两个下标组成一个数组一起返回
- 其中显然swap为交换数组两个部分
while(l < more) {
if(arr[l] < p) {
swap(arr,++less,l++);
} else if(arr[l] > p) {
swap(arr,--more,l);
} else {
l++;
}
}
return new int [] {
less+1,more -1};
- 第三返回的时候，返回的是界线后一位
我的代码
public class netherlanflag {
public static int [] partition(int [] arr,int l,int r,int p) {
int less = l - 1;
int more = r + 1;
while(l < more) {
if(arr[l] < p) {
swap(arr,++less,l++);
} else if(arr[l] > p) {
swap(arr,--more,l);
} else {
l++;
}
}
return new int [] {
less+1,more -1};
}
public static void swap(int []arr,int i, int j) {
int temp = arr[i];
arr[i]= arr[j];
arr[j] = temp;
}
}