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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163 | #include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <limits.h>
#include <errno.h>
// Quickselect in C
//
// PURPOSE
// Quickselect finds a zero-based order statistic without fully sorting the input.
//
// METHOD
// Partition around the middle-position pivot using a three-way partition. Retain only the
// side containing rank k, or return the pivot when k lies in the equal region.
//
// COMPLEXITY
// O(n) expected time for random input; O(n squared) worst case for this deterministic pivot.
// O(1) auxiliary space.
//
// CONTRACT
// 0 <= k < n. Mutates input. Returns the kth smallest value, counting duplicates.
// Public APIs infer array lengths except in C, which accepts pointer/length pairs. Vertex
// counts and matrix dimensions remain explicit. Arrays must contain signed 32-bit integers.
// Sorted order, correct tree topology, and caller-owned data-structure invariants are
// preconditions; checking them on every query would change query complexity.
// Scalar and shape checks are included. Value-range checks are included where needed to
// prevent invalid indexing or nontermination. Allocation sizes are capped at 4,000,000
// workspace entries. Arithmetic outside Python raises an error on 32-bit overflow; Python
// uses arbitrary precision. Native allocation failures propagate. In-place outputs may be
// partly updated after arithmetic failure. These are single-threaded, in-memory
// implementations.
//
// LANGUAGE DETAILS
// C99: checked calloc/free workspace, const input pointers, explicit size_t lengths. Errors
// return INT_MIN with errno=EINVAL, ERANGE, or ENOMEM. Check errno before using a result.
// Input/output arrays must not alias. No variable-length stack arrays.
//
// EXAMPLE
// Expected return: 5.
// Expected a after call: [2, 2, 5, 7, 9].
// Run: gcc -std=c99 -O2 example.c -o example && ./example
// Tests for this exact source artifact run before publication. The accompanying source
// archive includes unit cases, runtime results, and source hashes.
static int alg_min(int a, int b) { return a < b ? a : b; }
static int alg_max(int a, int b) { return a > b ? a : b; }
static int alg_checked(int64_t v, int *error) {
if (v < INT_MIN || v > INT_MAX) { *error = 1; return 0; }
return (int)v;
}
static int alg_add(int a, int b, int *e) { return alg_checked((int64_t)a+b,e); }
static int alg_sub(int a, int b, int *e) { return alg_checked((int64_t)a-b,e); }
static int alg_mul(int a, int b, int *e) { return alg_checked((int64_t)a*b,e); }
int quickselect(int *a, size_t a_length, int k) {
int alg_error = 0;
int alg_result = INT_MIN;
errno = 0;
if (a_length > 1000000 || (a_length && !a)) { errno = EINVAL; alg_result = INT_MIN; goto alg_cleanup; }
int n = (int)a_length;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (!(((n >= 0) && (n <= 1000000)))) { errno = EINVAL; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (!(((int)a_length == n))) { errno = EINVAL; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (!(((k >= 0) && (k < n)))) { errno = EINVAL; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int low = 0;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int high = alg_sub(n, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int left = 0;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int mid = 0;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int right = 0;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int pivot = 0;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
int temp = 0;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
while (1) {
int alg_condition = (low <= high);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (!alg_condition) break;
pivot = a[alg_add(low, (alg_sub(high, low, &alg_error) / 2), &alg_error)];
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
left = low;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
mid = low;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
right = high;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
while (1) {
int alg_condition = (mid <= right);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (!alg_condition) break;
int alg_condition_1 = (a[mid] < pivot);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_condition_1) {
temp = a[left];
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
a[left] = a[mid];
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
a[mid] = temp;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
left = alg_add(left, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
mid = alg_add(mid, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
} else {
int alg_condition_2 = (a[mid] > pivot);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_condition_2) {
temp = a[mid];
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
a[mid] = a[right];
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
a[right] = temp;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
right = alg_sub(right, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
} else {
mid = alg_add(mid, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
}
}
}
int alg_condition_3 = (k < left);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_condition_3) {
high = alg_sub(left, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
} else {
int alg_condition_4 = (k > right);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
if (alg_condition_4) {
low = alg_add(right, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
} else {
alg_result = pivot;
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
goto alg_cleanup;
}
}
}
alg_result = alg_sub(0, 1, &alg_error);
if (alg_error) { errno = ERANGE; alg_result = INT_MIN; goto alg_cleanup; }
goto alg_cleanup;
alg_cleanup:
return alg_result;
}
int main(void) {
int a[] = {7, 2, 9, 2, 5};
int k = 2;
int result = quickselect(a, 5, k);
printf("%d\n", result);
// Expected a: [2, 2, 5, 7, 9]
for (int alg_print = 0; alg_print < 5; alg_print++) printf("%d ", a[alg_print]);
printf("\n");
if (errno) { perror("algorithm"); return 1; }
return 0;
}
|