forked from JuliaLang/julia
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmedian_MU.jl
54 lines (48 loc) · 903 Bytes
/
median_MU.jl
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
function median_MU(S)
n = length(S)
if n < 1500
C = sort(S)
half = iround(n/2)
if mod(n,2) == 1
m = C[half]
else
m = (C[half]+C[half+1])/2
end
return m
end
# finds median of set S, a la Mitzenmacher & Upfal, Alg. 3.1
n75 = int64(ceil(n^.75))
R = S[int64(ceil(n*rand(n75)))]
sort!(R)
d = R[max(1.0, floor((n75/2)-sqrt(n)))]
u = R[ceil((n75/2)+sqrt(n))]
C = similar(S, 0)
Ld = 0
Lu = 0
for i = 1:n
if S[i] < d
Ld = Ld + 1
else
if S[i] > u
Lu = Lu + 1
else
C = push(C,S[i])
end
end
end
if (Ld > n/2) || (Lu > n/2)
m = median(S)
else
if length(C) > 4*n75
m = median(S)
else
sort!(C)
if mod(n,2) == 1
m = C[floor(n/2)-Ld+1]
else
m = (C[floor(n/2)-Ld]+C[floor(n/2)-Ld+1])/2
end
end
end
return m
end