Created
September 17, 2013 21:17
-
-
Save wieslawsoltes/6600771 to your computer and use it in GitHub Desktop.
Solution for http://ayende.com/blog/163394/new-interview-question using ParallelSort with 10 million items.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
using System; | |
using System.Collections.Generic; | |
using System.Linq; | |
using System.Text; | |
using System.Threading; | |
using System.Threading.Tasks; | |
namespace Demo | |
{ | |
static class Program | |
{ | |
public static IEnumerable<long> GetRuns(this int[] n) | |
{ | |
//Array.Sort(n); | |
int pos = 0; | |
while (pos < n.Length - 1) | |
{ | |
int i, j, first = n[pos]; | |
for (j = pos + 1; j < n.Length; j++) | |
{ | |
if ((i = n[j]) == first + 1) | |
first = i; | |
else | |
break; | |
} | |
yield return Pack(pos, j); | |
pos = j; | |
} | |
} | |
private static long Pack(int start, int end) | |
{ | |
long run = end; | |
run = run << 32; | |
run = run | (uint)start; | |
return run; | |
} | |
private static void Unpack(long run, out int start, out int end) | |
{ | |
start = (int)(run & uint.MaxValue); | |
end = (int)(run >> 32); | |
} | |
static void Main(string[] args) | |
{ | |
//Test1(); | |
Test2(); | |
Console.ReadLine(); | |
} | |
private static void Print(int[] n, long run) | |
{ | |
bool first = true; | |
Console.Write('{'); | |
int start; | |
int end; | |
Unpack(run, out start, out end); | |
for (int i = start; i < end; i++) | |
{ | |
if (!first) | |
Console.Write(','); | |
else | |
first = false; | |
Console.Write(n[i]); | |
} | |
Console.WriteLine('}'); | |
} | |
private static void Test1() | |
{ | |
int[] n = { 1, 59, 12, 43, 4, 58, 5, 13, 46, 3, 6 }; | |
Array.Sort(n); | |
var runs = n.GetRuns(); | |
foreach (var run in runs) | |
Print(n, run); | |
} | |
private static void DummyPrint(int[] n, long run) | |
{ | |
//int start; | |
//int end; | |
//Unpack(run, out start, out end); | |
} | |
private static void Test2() | |
{ | |
const int size = 10 * 1024 * 1024; | |
int[] n = new int[size]; | |
var rand = new Random(); | |
for (int i = 0; i < size; i++) | |
n[i] = rand.Next(); | |
var sw = System.Diagnostics.Stopwatch.StartNew(); | |
//Array.Sort(n); | |
ParallelSort.Sort.ParallelQuickSort<int>(n); | |
sw.Stop(); | |
Console.WriteLine(sw.Elapsed.TotalMilliseconds + "ms"); | |
sw.Restart(); | |
var runs = n.GetRuns(); | |
foreach (var run in runs) | |
DummyPrint(n, run); | |
sw.Stop(); | |
Console.WriteLine(sw.Elapsed.TotalMilliseconds + "ms"); | |
} | |
} | |
} | |
namespace ParallelSort | |
{ | |
#region Parallel Sort | |
public static class Sort | |
{ | |
public static int Threshold = 150; // array length to use InsertionSort instead of SequentialQuickSort | |
public static void InsertionSort<T>(T[] array, int from, int to) where T : IComparable<T> | |
{ | |
for (int i = from + 1; i < to; i++) | |
{ | |
var a = array[i]; | |
int j = i - 1; | |
//while (j >= from && array[j] > a) | |
while (j >= from && array[j].CompareTo(a) == -1) | |
{ | |
array[j + 1] = array[j]; | |
j--; | |
} | |
array[j + 1] = a; | |
} | |
} | |
static void Swap<T>(T[] array, int i, int j) where T : IComparable<T> | |
{ | |
var temp = array[i]; | |
array[i] = array[j]; | |
array[j] = temp; | |
} | |
static int Partition<T>(T[] array, int from, int to, int pivot) where T : IComparable<T> | |
{ | |
// Pre: from <= pivot < to (other than that, pivot is arbitrary) | |
var arrayPivot = array[pivot]; // pivot value | |
Swap(array, pivot, to - 1); // move pivot value to end for now, after this pivot not used | |
var newPivot = from; // new pivot | |
for (int i = from; i < to - 1; i++) // be careful to leave pivot value at the end | |
{ | |
// Invariant: from <= newpivot <= i < to - 1 && | |
// forall from <= j <= newpivot, array[j] <= arrayPivot && forall newpivot < j <= i, array[j] > arrayPivot | |
//if (array[i] <= arrayPivot) | |
if (array[i].CompareTo(arrayPivot) != -1) | |
{ | |
Swap(array, newPivot, i); // move value smaller than arrayPivot down to newpivot | |
newPivot++; | |
} | |
} | |
Swap(array, newPivot, to - 1); // move pivot value to its final place | |
return newPivot; // new pivot | |
// Post: forall i <= newpivot, array[i] <= array[newpivot] && forall i > ... | |
} | |
public static void SequentialQuickSort<T>(T[] array) where T : IComparable<T> | |
{ | |
SequentialQuickSort(array, 0, array.Length); | |
} | |
static void SequentialQuickSort<T>(T[] array, int from, int to) where T : IComparable<T> | |
{ | |
if (to - from <= Threshold) | |
{ | |
InsertionSort<T>(array, from, to); | |
} | |
else | |
{ | |
int pivot = from + (to - from) / 2; // could be anything, use middle | |
pivot = Partition<T>(array, from, to, pivot); | |
// Assert: forall i < pivot, array[i] <= array[pivot] && forall i > ... | |
SequentialQuickSort(array, from, pivot); | |
SequentialQuickSort(array, pivot + 1, to); | |
} | |
} | |
public static void ParallelQuickSort<T>(T[] array) where T : IComparable<T> | |
{ | |
ParallelQuickSort(array, 0, array.Length, | |
(int)Math.Log(Environment.ProcessorCount, 2) + 4); | |
} | |
static void ParallelQuickSort<T>(T[] array, int from, int to, int depthRemaining) where T : IComparable<T> | |
{ | |
if (to - from <= Threshold) | |
{ | |
InsertionSort<T>(array, from, to); | |
} | |
else | |
{ | |
int pivot = from + (to - from) / 2; // could be anything, use middle | |
pivot = Partition<T>(array, from, to, pivot); | |
if (depthRemaining > 0) | |
{ | |
Parallel.Invoke( | |
() => ParallelQuickSort(array, from, pivot, depthRemaining - 1), | |
() => ParallelQuickSort(array, pivot + 1, to, depthRemaining - 1)); | |
} | |
else | |
{ | |
ParallelQuickSort(array, from, pivot, 0); | |
ParallelQuickSort(array, pivot + 1, to, 0); | |
} | |
} | |
} | |
} | |
#endregion | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment