我下面的荷兰国旗解决方案似乎不适用于仅包含 3 个元素(0、1 和 2)的给定输入数组。
如果我减小数组的大小,它会起作用。我无法识别错误。
我错过了什么吗?
class DNF{
public static void sort(int [] arr, int size, int low, int high) {
if(size == 0)
return;
int lower = 0;
int upper = size - 1;
while(lower < size && arr[lower] == low)
lower++;
while(upper >=0 && arr[upper] == high)
upper--;
int temp = 0;
int pivot;
for(pivot = lower; pivot <= upper; ) {
if(arr[pivot] == low) {
temp = arr[pivot];
arr[pivot] = arr[lower];
arr[lower] = temp;
pivot++;
lower++;
} else if(arr[pivot] == high) {
temp = arr[pivot];
arr[pivot] = arr[upper];
arr[upper] = temp;
pivot++;
upper--;
} else {
pivot++;
}
}
}
public static void main(String [] args) {
int arr [] = {0,1,2,1,2,0,1,1,1,0,0,0,1,0,2,1};
for(int i=0; i<arr.length; i++) {
System.out.print(arr[i]+" "); //0 1 2 1 2 0 1 1 1 0 0 0 1 0 2 1
}
sort(arr, arr.length, 0, 2);
System.out.println();
for(int i=0; i<arr.length; i++) {
System.out.print(arr[i]+" "); // 0 0 0 0 0 0 1 1 1 1 1 2 1 1 2 2
}
}
}
更新:上面相同的代码适用于较小的数组:http: //ideone.com/bgEgCs