9-9 荷兰国旗问题:设有一个仅由红、白、蓝3种颜色的条块

#include <stdio.h>
#include <stdlib.h>
//只分两段的荷兰国旗问题 
void swap(int *a,int i,int j)
{
	int temp=a[i];
	a[i]=a[j];
	a[j]=temp;
}
void middle(int *a,int left,int right,int sum)
{
	int i=left-1;
	int current=left;
	int j=right;
	while(current<=right)
	{
		if(a[current]<sum)
		{
			swap(a,i+1,current);
			i++;
			current++;
		}
		else
			current++;
	}
}
//分三段的荷兰国旗问题
void three_middle(int *a,int left,int right,int number)
{
	int i=left-1;
	int current=left;
	int j=right+1;
	
	int tag;
	while(current<j)
	{
		if(a[current]<number)
		{
			swap(a,i+1,current);
			i++;
			current++;
//			printf("执行第1步\n");
//			for(tag=0;tag<=right;tag++)
//			printf("%d ",a[tag]);
//			printf("\n");
		}
		else if(a[current]==number)
		{
			current++;
//			printf("执行第2步\n");
//				for(tag=0;tag<=right;tag++)
//			printf("%d ",a[tag]);
//			printf("\n");
		}
		else
		{
			swap(a,j-1,current);
			j--;
//			printf("执行第3步\n");
//				for(tag=0;tag<=right;tag++)
//			printf("%d ",a[tag]);
//			printf("\n");
		}
		
	}
 } 
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		int *a;
		a=(int *)malloc(sizeof(int)*n);
		int i;
		for(i=0;i<n;i++)
		scanf("%d",&a[i]);
		int k;
		scanf("%d",&k);
		three_middle(a,0,n-1,k);
			for(i=0;i<n;i++)
		printf("%d ",a[i]);
	}

	return 0;
}

public class Main{
    //for test
    public static void main(String[] args) {
        int[] arr = {1,6,3,5,6,7,6,9,2,4,6,8};
        int[] ret=netherlandsFlag(arr,0,arr.length-1);
        System.out.println(Arrays.toString(ret));
    }
    public static int[] netherlandsFlag(int[] arr,int L,int R){
        if(L>R){
            return new int[]{-1,-1};
        }
        if(L==R){
            return new int[]{L,R};
        }
        int less=L-1;
        int right=R;
        int index=L;
        while(index<R){
            if(arr[index]==arr[R]){
                index++;
            }else if(arr[index]<arr[R]){
                swap(arr,index++,++less);
            }else{
                swap(arr,index,--more);
            }
        }
        swap(arr,more,R);
        return new int[]{less+1,more};
    }
    public static void swap(int[] arr,int a,int b){
        int tmp=arr[a];
        arrr[a]=arr[b];
        arr[b]=tmp;
    }
}