編程演算法 - 排序分割 代碼(C)

標籤：mystra   編程演算法   排序分割   快速排序   代碼   

排序分割 代碼(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 

編程演算法 - 排序分割 代碼(C)