"Maximum Element After Decreasing and Rearranging" solution

src/maximum_element_after_decreasing_and_rearranging.py

@@ -0,0 +1,15 @@
+class Solution:
+    def maximumElementAfterDecrementingAndRearranging(
+        self, arr: list[int]
+    ) -> int:
+        arr = sorted(arr)
+
+        last = 0
+
+        for x in arr:
+            if x > last + 1:
+                last = last + 1
+            else:
+                last = x
+
+        return last

tests/test_maximum_element_after_decreasing_and_rearranging.py

@@ -0,0 +1,18 @@
+import pytest
+
+from src.maximum_element_after_decreasing_and_rearranging import Solution
+
+
+@pytest.mark.parametrize(
+    "arr,expected",
+    (
+        ([2, 2, 1, 2, 1], 2),
+        ([100, 1, 1000], 3),
+        ([1, 2, 3, 4, 5], 5),
+    ),
+)
+def test_solution(arr, expected):
+    assert (
+        Solution().maximumElementAfterDecrementingAndRearranging(arr)
+        == expected
+    )