diff options
Diffstat (limited to 'challenge-244/lubos-kolouch/python/ch-1.py')
| -rw-r--r-- | challenge-244/lubos-kolouch/python/ch-1.py | 25 |
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] |
