算法练习bat----------04荷兰国旗问题

给定一个无序数字，小于num放左边，等于num放中间，大于num放右边，要求空间复制度O(1)，时间复杂度O（n）

实际上就三种情况：（使用less表示数组-1，more表示数组length，cur表示当前位置）

① 当数字小于num，less的下一个和cur交换，cur++(也就是下面的swap(arr, ++less, l++);)

②当数字等于num，cur++

③当数字大于num，more的上前一个和cur交换，cur不动（因为换过来的数字大小不确定，所以下表不能移动），（也就是下面的swap(arr, --more, l);）

 

结束条件：当more和cur相遇

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 };
	}

	// for test
	public static void swap(int[] arr, int i, int j) {
		int tmp = arr[i];
		arr[i] = arr[j];
		arr[j] = tmp;
	}
}