荷兰国旗问题

已知一个整型数组arr，和一个整数num，请把小于num的数放在数组的左边，等于num的数放在数组的中间，大于num的数放在数组的右边。 要求：时间复杂度为O(N)，额外空间复杂度O(1)。

 

这是属于快排的一种思想

 

基础：给一个数r，如何把小于等于的数放在左边，把大于的数放在右边

假设有一个数组 arr-> 6 5 3 5 7 2     则设小于等于设一个左边界 ，左边界里的数都小于等于r

以arr数组为例，起始左边界L在-1位置，然后i从下标0开始依次跟r比较，如果小于等于r则与L+1位置的数交换，L扩一个位置；如果大于r，则i继续下移，直到把数组的数遍历完

 进阶就是荷兰国旗问题

还是假设有一个数组 arr-> 6 5 3 5 7 2     设一个左边界L，一个右边界R，，，arr数组的左边界L起始在-1位置，右边界起始在n位置

从下标为0的位置开始遍历，如果i位置的数小于r，则与L+1位置的数交换，L扩一个位置，如果等于r，则i下移一个位置，如果大于r，则与R前一个位置的数x交换，R前移一个位置，交换后i不下移，因为x是未知的，所以需要继续与r比大小，比的规则如上，直到i与R相撞，遍历结束。

分析：当i位置的数比r小，i后移，L后移，R不动

           当i位置的数等于r，i后移，L和R不动

           当i位置的数大于r，i和不动，R前移，，，所有上面三种情况分析，i和R是一定会相撞的

代码实现

c语言

#include <stdio.h>
#include <stdlib.h>
int main()
{
    void partition(int arr[],int L,int R,int p);
    int n,i,num;
    scanf("%d",&n);
    int arr[n];
    for(i=0;i<n;i++)
        scanf("%d",&arr[i]);
    scanf("%d",&num);
    partition(arr,0,n-1,num);
    for(i=0;i<n;i++)
    printf("%d ",arr[i]);
    return 0;
}
void partition(int arr[],int L,int R,int p)
{
    void swap(int arr[],int i,int j);
    int less=L-1;
    int more=R+1;
    int index=L;
    while(index<more)
    {
        if(arr[index]<p){
            swap(arr,++less,index++);}
            else if(arr[index]>p){
                swap(arr,--more,index);}
            else {
                    index++;}
    }
}
void swap(int arr[],int i,int j)
{
    int t;
    t=arr[i];
    arr[i]=arr[j];
    arr[j]=t;
}

java

public class NetherlandsFlag {

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

    // for test
    public static int[] generateArray() {
        int[] arr = new int[10];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = (int) (Math.random() * 3);
        }
        return arr;
    }

    // for test
    public static void printArray(int[] arr) {
        if (arr == null) {
            return;
        }
        for (int i = 0; i < arr.length; i++) {
            System.out.print(arr[i] + " ");
        }
        System.out.println();
    }

    public static void main(String[] args) {
        int[] test = generateArray();

        printArray(test);
        int[] res = partition(test, 0, test.length - 1, 1);
        printArray(test);
        System.out.println(res[0]);
        System.out.println(res[1]);

    }
}