1 | /* |
2 | * Copyright (c) 2011, 2019, Oracle and/or its affiliates. All rights reserved. |
3 | * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
4 | * |
5 | * This code is free software; you can redistribute it and/or modify it |
6 | * under the terms of the GNU General Public License version 2 only, as |
7 | * published by the Free Software Foundation. |
8 | * |
9 | * This code is distributed in the hope that it will be useful, but WITHOUT |
10 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
11 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
12 | * version 2 for more details (a copy is included in the LICENSE file that |
13 | * accompanied this code). |
14 | * |
15 | * You should have received a copy of the GNU General Public License version |
16 | * 2 along with this work; if not, write to the Free Software Foundation, |
17 | * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
18 | * |
19 | * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
20 | * or visit www.oracle.com if you need additional information or have any |
21 | * questions. |
22 | * |
23 | */ |
24 | |
25 | #ifndef SHARE_UTILITIES_QUICKSORT_HPP |
26 | #define SHARE_UTILITIES_QUICKSORT_HPP |
27 | |
28 | #include "memory/allocation.hpp" |
29 | #include "runtime/globals.hpp" |
30 | #include "utilities/debug.hpp" |
31 | |
32 | class QuickSort : AllStatic { |
33 | |
34 | private: |
35 | template<class T> |
36 | static void swap(T* array, size_t x, size_t y) { |
37 | T tmp = array[x]; |
38 | array[x] = array[y]; |
39 | array[y] = tmp; |
40 | } |
41 | |
42 | // As pivot we use the median of the first, last and middle elements. |
43 | // We swap in these three values at the right place in the array. This |
44 | // means that this method not only returns the index of the pivot |
45 | // element. It also alters the array so that: |
46 | // array[first] <= array[middle] <= array[last] |
47 | // A side effect of this is that arrays of length <= 3 are sorted. |
48 | template<class T, class C> |
49 | static size_t find_pivot(T* array, size_t length, C comparator) { |
50 | assert(length > 1, "length of array must be > 0" ); |
51 | |
52 | size_t middle_index = length / 2; |
53 | size_t last_index = length - 1; |
54 | |
55 | if (comparator(array[0], array[middle_index]) > 0) { |
56 | swap(array, 0, middle_index); |
57 | } |
58 | if (comparator(array[0], array[last_index]) > 0) { |
59 | swap(array, 0, last_index); |
60 | } |
61 | if (comparator(array[middle_index], array[last_index]) > 0) { |
62 | swap(array, middle_index, last_index); |
63 | } |
64 | // Now the value in the middle of the array is the median |
65 | // of the fist, last and middle values. Use this as pivot. |
66 | return middle_index; |
67 | } |
68 | |
69 | template<bool idempotent, class T, class C> |
70 | static size_t partition(T* array, size_t pivot, size_t length, C comparator) { |
71 | size_t left_index = 0; |
72 | size_t right_index = length - 1; |
73 | T pivot_val = array[pivot]; |
74 | |
75 | for ( ; true; ++left_index, --right_index) { |
76 | for ( ; comparator(array[left_index], pivot_val) < 0; ++left_index) { |
77 | assert(left_index < length, "reached end of partition" ); |
78 | } |
79 | for ( ; comparator(array[right_index], pivot_val) > 0; --right_index) { |
80 | assert(right_index > 0, "reached start of partition" ); |
81 | } |
82 | |
83 | if (left_index < right_index) { |
84 | if (!idempotent || comparator(array[left_index], array[right_index]) != 0) { |
85 | swap(array, left_index, right_index); |
86 | } |
87 | } else { |
88 | return right_index; |
89 | } |
90 | } |
91 | |
92 | ShouldNotReachHere(); |
93 | return 0; |
94 | } |
95 | |
96 | template<bool idempotent, class T, class C> |
97 | static void inner_sort(T* array, size_t length, C comparator) { |
98 | if (length < 2) { |
99 | return; |
100 | } |
101 | size_t pivot = find_pivot(array, length, comparator); |
102 | if (length < 4) { |
103 | // arrays up to length 3 will be sorted after finding the pivot |
104 | return; |
105 | } |
106 | size_t split = partition<idempotent>(array, pivot, length, comparator); |
107 | size_t first_part_length = split + 1; |
108 | inner_sort<idempotent>(array, first_part_length, comparator); |
109 | inner_sort<idempotent>(&array[first_part_length], length - first_part_length, comparator); |
110 | } |
111 | |
112 | public: |
113 | // The idempotent parameter prevents the sort from |
114 | // reordering a previous valid sort by not swapping |
115 | // fields that compare as equal. This requires extra |
116 | // calls to the comparator, so the performance |
117 | // impact depends on the comparator. |
118 | template<class T, class C> |
119 | static void sort(T* array, size_t length, C comparator, bool idempotent) { |
120 | // Switch "idempotent" from function paramter to template parameter |
121 | if (idempotent) { |
122 | inner_sort<true>(array, length, comparator); |
123 | } else { |
124 | inner_sort<false>(array, length, comparator); |
125 | } |
126 | } |
127 | }; |
128 | |
129 | |
130 | #endif // SHARE_UTILITIES_QUICKSORT_HPP |
131 | |