ALDS1_6_B: Partition

题目链接：http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ALDS1_6_B

输入一个数组

输出经过经过一次分割之后的数组，并且标注出参考元素；

 

思路：一个数组，我们要从给[left,right]排序，我们先初始化i=left-1,j=left;

先看A[i+1]和A[right]比较：如果小于，就i++，j++；如果大于或者等于，就i不变，j++；

直到j=right-1为止，然后把A[i+1]和A[right]换位置，这样就实现了，A[left]-A[i]都是小于A[i+1]的，A[i+2]-A[right]是大于A[i+1]的。

代码如下：

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;

int A[100010];
int n;

int partition(int left,int right){
	int i=left-1;
	for(int j=left;j<right;j++){
		if(A[j]<=A[right]) {
			swap(A[i+1],A[j]);
			i++;
		}
	}
	swap(A[right],A[i+1]);
	return i+1;	
}

int main (){
	cin>>n;
	for(int i =0;i<n;i++) cin>>A[i];
	int x=partition(0,n-1);
	for(int i =0;i<n;i++) {
		if(i==0) cout<<A[i];
		else if(i==x) cout<<" ["<<A[i]<<"]";
		else cout<<" "<<A[i];
	}
	cout<<endl;
	return 0;
}

错点：

 

1 if(A[j]<=A[right])要加等号，让等于的情况去参考值的左边

2 输出格式要注意加"[ ]"