荷兰国旗问题

题目描述：

给定一个数组arr，和一个数num，请把小于num的数放在数组的 左边，等于num的数放在数组的中间，大于num的数放在数组的右边

package lesson01;

import java.util.Random;

public class demo03_NetherlandsFlag {

  public static void main(String[] args) {

    int n = 10; // 生成随机数组的长度
    int[] arr = generateRandomArray(n); // 生成随机数组
    printArray(arr); // 打印

    int num = 5; // 给定的数
    int l = 0, r = arr.length - 1;

    int[] a = partition(arr, l, r, num); // 排序
    System.out.println();
    printArray(a);
  }

  public static int[] partition(int[] arr, int l, int r, int num) { // 排序
    int curlen = l; // 当前要比较的数
    int less = l - 1; // 小于num的范围，初始界限
    int more = r + 1; // 大于num的范围，初始界限
    while (curlen < more) {
      if (arr[curlen] < num) {
        swap(arr, ++less, curlen++); // 小于num的范围的指针，先向后移动一位 ，待swap（）函数执行完后，curlen在++
      } else if (arr[curlen] > num) {
        swap(arr, --more, curlen); // 大于num的范围的指针，先向前移动一位 ，待swap（）函数执行完后，
                      // curlen不移动，交换过来的数继续和num进行比较
      } else {
        curlen++;
      }
    }
    return arr;
  }

  public static void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
  }

  public static void printArray(int[] arr) { // 打印
    for (int i = 0; i < arr.length; i++)
      System.out.print(" " + arr[i]);
  }

  public static int[] generateRandomArray(int n) { // 生成随机数组
    Random r = new Random();
    int arr[] = new int[n];
    for (int i = 0; i < n; i++)
      arr[i] = (int) (r.nextInt(n) + 1);
    return arr;
  }

}