【算法-Java实现】将数组按某值划分成左边小，中间相等，右边大的形式

【算法-Java实现】将数组按某值划分成左边小，中间相等，右边大的形式

一.问题描述：

1.输入：一个无序整型数组arr，其中arr中元素不唯一。

如：int[] arr = { 3, 4, 6, 9, 5, 5, 8, 8, 2, 1, 7, 7, 0 };

2.输出：将数组arr按某值划分成左边小，中间相等，右边大的形式，并输出。

二.问题解答：

此问题的另外一种问法：荷兰国旗问题

有一个数组，只有红球，蓝球，黄球，请将红球放在数组左边，蓝球放在数组中间，黄球放在数组右边。

思路：

1.首先用一个基准num作为中间值，cur指针从下标0依次遍历数组至下标arr.length-1。

2.less指针一开始指向-1，more指针指向arr.length。

3.cur指针依次移动，判断当前cur下标元素和num值之间的大小关系，进行元素交换。

具体实现见代码，结果如下：

三.算法分析：

1.时间复杂度为O(N)，相当于遍历数组。

2.额外空间复杂度为O(1),无需使用额外空间。

代码如下

import java.util.Arrays;
/*
 * 问题：将数组按某值划分成左边小，中间相等，右边大的形式
 * 
 */
public class ArraySort {
	//测试
	public static void main(String[] args) {
		int[] arr = { 3, 4, 6, 9, 5, 5, 8, 8, 2, 1, 7, 7, 0 };
		int[] newArr = partition(arr, 0, arr.length - 1, 5);
		System.out.println(Arrays.toString(newArr));
	}
    //算法实现
	public static int[] partition(int[] arr, int L, int R, int num) {
		int less = -1;
		int more = R + 1;
		int cur = L;
		while (cur < more) {
			if (arr[cur] > num) {
				swap(arr, --more, cur);
			} else if (arr[cur] < num) {
				swap(arr, ++less, cur++);
			} else {
				cur++;
			}
		}
		return arr;
	}
	//元素交换
	public static void swap(int[] arr, int i, int j) {
		int temp = arr[i];
		arr[i] = arr[j];
		arr[j] = temp;
	}
}