-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathradixSort.cpp
90 lines (66 loc) · 1.58 KB
/
radixSort.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include <iostream>
#include <fstream>
#include <vector>
#include <chrono>
#include <string>
#include <algorithm>
#include <random>
#include <memory>
#define BYTE_MASK 0xFF
#define BYTE_SIZE 0xFF
using namespace std;
void CountingSort(int *arr, int *outArr, int nElements, int maskByte)
{
int byteOffset = 8 * maskByte;
int mask = BYTE_MASK << byteOffset;
if (maskByte == 3)
mask &= ~(1UL << 32);
long long countArr[BYTE_SIZE + 1 ] = {0};
for (int i = 0; i < nElements; i++)
{
int v = arr[i] & mask;
v = v >> byteOffset;
countArr[v]++;
}
for (int i = 1; i <= BYTE_SIZE; i++)
countArr[i] += countArr[i - 1];
for (int i = nElements - 1; i >= 0; i--)
{
int countIndex = (arr[i] & mask) >> byteOffset;
outArr[(countArr[countIndex]--) - 1] = arr[i];
}
}
void RadixSort(int *arr, int nElements)
{
int *tempArr = (int *)malloc(nElements * sizeof(int));
for (int i = 0; i < sizeof(int); i++)
{
CountingSort(arr, tempArr, nElements, i);
memcpy(arr, tempArr, nElements * sizeof(int));
}
free(tempArr);
}
int main()
{
std::random_device dev;
std::mt19937 rng(dev());
std::uniform_int_distribution<std::mt19937::result_type> dist(1, 0xffffffff);
int nSize = 10000000;
int *arr = (int *)malloc(sizeof(int) * nSize);
for (int i = 0; i < nSize; i++)
{
arr[i] = dist(rng);
}
RadixSort(arr, nSize);
ofstream output("output.txt");
if(!output.is_open())
{
free(arr);
return -1;
}
for(int i = 0; i < nSize; i++)
output << arr[i] << '\n';
output.close();
free(arr);
return 0;
}