aboutsummaryrefslogtreecommitdiff
path: root/challenge-244/lubos-kolouch/python/ch-1.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-244/lubos-kolouch/python/ch-1.py')
-rw-r--r--challenge-244/lubos-kolouch/python/ch-1.py25
1 files changed, 25 insertions, 0 deletions
diff --git a/challenge-244/lubos-kolouch/python/ch-1.py b/challenge-244/lubos-kolouch/python/ch-1.py
new file mode 100644
index 0000000000..50799f5339
--- /dev/null
+++ b/challenge-244/lubos-kolouch/python/ch-1.py
@@ -0,0 +1,25 @@
+from typing import List
+
+
+def count_smaller(nums: list[int]) -> list[int]:
+ """
+ Count the number of elements smaller than the current element at each index.
+
+ Args:
+ nums (List[int]): An array of integers.
+
+ Returns:
+ List[int]: An array containing the counts.
+ """
+ n = len(nums)
+ result = []
+ for i in range(n):
+ count = sum(1 for j in range(n) if nums[j] < nums[i] and j != i)
+ result.append(count)
+ return result
+
+
+# Tests
+assert count_smaller([8, 1, 2, 2, 3]) == [4, 0, 1, 1, 3]
+assert count_smaller([6, 5, 4, 8]) == [2, 1, 0, 3]
+assert count_smaller([2, 2, 2]) == [0, 0, 0]