编程算法 - 排序分割 代码(C)

排序分割 代码(C)

本文地址: http://blog.csdn.net/caroline_wendy

排序分割, 把一个数组分为, 大于k\小于k\等于k的三个部分.

可以使用快速排序的Partition函数, 进行处理, 把大于k的放在左边, 小于k的放在右边.

使用一个变量记录中间的位置, 则时间复杂度为O(3n/2).

代码:

/*
 * main.cpp
 *
 *  Created on: 2014.9.18
 *      Author: Spike
 */

/*eclipse cdt, gcc 4.8.1*/

#include <iostream>
#include <list>
#include <queue>
#include <string.h>
#include <stdlib.h>

using namespace std;

int RandomInRange(int min, int max) {
	return rand()%(max-min+1) + min;
}

void Swap(int* num1, int* num2) {
	int tmp = *num1;
	*num1 = *num2;
	*num2 = tmp;
}

int PartitionK (int data[], int length, int k)
{
	if (data == NULL || length <= 0)
		return -1;
	int small = -1;
	int index = 0;
	for (index=0; index<length; ++index) {
		if (data[index] < k) {
			small++;
			if (small != index)
				Swap(&data[small], &data[index]);
		}
	}

	int pos = ++small; //记录位置

	small = length;
	index = 0;
	for (index = length-1; index>=pos; index--) {
		if (data[index] > k) {
			small--;
			if (small != index)
				Swap(&data[small], &data[index]);
		}
	}
	return --small;
}

int main (void)
{
	int data[] = {1, 3, 2, 5, 2, 0, 5, 2, -1, 0, 4};
	int length = sizeof(data)/sizeof(data[0]);
	PartitionK(data, length, 2);
	for (int i=0; i<length; ++i) {
		cout << data[i] << " ";
	}
	cout << endl;
	return 0;
}

输出:

1 0 -1 0 2 2 2 3 5 5 4