From 2067a01f422224d81ef87eaa5409f0a5116bb051 Mon Sep 17 00:00:00 2001
From: Nuwan Yapa <yapanuwan@gmail.com>
Date: Mon, 4 Mar 2024 13:41:21 -0500
Subject: [PATCH] Better sum for complex arrays

---
 helper.jl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/helper.jl b/helper.jl
index 420c819..ce9f7e6 100644
--- 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
\ No newline at end of file
+end
+
+better_sum(arr::Array{ComplexF64}) = better_sum(real.(arr)) + 1im * better_sum(imag.(arr))