diff --git a/src/main.rs b/src/main.rs
index aefb070..a339718 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -22,7 +22,7 @@ fn compute_cost(crabs: &[i64], mid: i64) -> i64 {
     for &crab in crabs {
         let distance = (crab - mid).abs();
 
-        cost += distance;
+        cost += distance * (distance + 1) / 2;
     }
     cost
 }