三色球排序

#include <stdio.h>
#define BLUE -1
#define RED 0
#define YELLOW 1

void swap(int *a,int *b){
    int t=*a;
    *a=*b;
    *b=t;
}

void MySort( int A[], int N ){
    int head=0,end=N-1;
    for(int i=0;i<end;){
        if(A[i]==YELLOW){
            swap(&A[i],&A[head]);
            head++;
        }
        else if(A[i]==RED){
            i++;
        }
        else if(A[i]==BLUE){
            swap(&A[i],&A[end]);
            end--;
        }
    }
}

int main(){
    int N;
    int A[10];
    scanf("%d", &N);
    for(int i=0;i<N;i++){
        scanf("%d", &A[i]);
    }
    MySort(A,N);
    for(int i=0;i<N;i++){
        printf("%d ", A[i]);
    }
    return 0;
}