Aizu Online Judge in C++ #ALDS1_6_B : Partition

Aizu Online Judge（AOJ）の過去問から、その提出コードの解答例です。

・問題 "Partition"
https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_6_B
partition ( A, p, r )は、配列 A[ p..r ] を A[ p..q − 1] の各要素が A[q] 以下で、A[ q +1.. r ] の各要素が A[ q ] より大きい A[ p..q − 1] と A[q + 1..r ] に分割し、インデックス q を戻り値として返します。
数列 A を読み込み、次の疑似コードに基づいた partition を行うプログラムを作成してください。

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #ALDS1_6_B : Partition

/*
Aizu Online Judge in C++ #ALDS1_6_B : Partition
https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_6_B
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int partition(int* a,int p,int r){
    int x=a[r];
    int i=p-1;
    for(int j=p;j<r;j++){
        if(a[j]<=x){
            i+=1;
            swap(a[i],a[j]);
        }
    }
    swap(a[i+1],a[r]);
    return i+1;
}
int main(void){
    int n;
    cin>>n;
    int a[n];
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    int p=partition(a,0,n-1);
    for(int i=0;i<n;i++){
        if(i==p){
            cout<<"["<<a[i]<<"]";
        }else{
            cout<<a[i];
        }
        cout<<((i==n-1) ? "\n" : " ");
    }
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge（AOJ）」です。
http://judge.u-aizu.ac.jp/onlinejudge/