Better sum for complex arrays

2024-03-04 13:41:21 -05:00 · 2024-03-04 13:41:21 -05:00 · 2067a01f42
parent bfb9c3d3b9
commit 2067a01f42
1 changed files with 3 additions and 1 deletions
--- a/helper.jl
+++ b/helper.jl
@ -7,4 +7,6 @@ function better_sum(arr::Array{Float64})
    sort!(neg_arr, rev=true)

    return sum(pos_arr) + sum(neg_arr)
-end
+end
+
+better_sum(arr::Array{ComplexF64}) = better_sum(real.(arr)) + 1im * better_sum(imag.(arr))