Merge pull request #1892 from CodingTestStudy2/최원준

[최원준] Day25

Author: github-actions[bot]

leetcode3/최원준/315. Count of Smaller Numbers After Self.py

@@ -0,0 +1,46 @@
+class Solution:
+    def countSmaller(self, nums: List[int]) -> List[int]:
+        n = len(nums)
+        ans = [0] * n
+
+        arr = [(num, i) for i, num in enumerate(nums)]
+
+        def merge_sort(arr):
+            if len(arr) <= 1:
+                return arr
+
+            mid = len(arr) // 2
+
+            left = merge_sort(arr[:mid])
+            right = merge_sort(arr[mid:])
+
+            merged = []
+
+            i = 0
+            j = 0
+            smaller = 0
+
+            while i < len(left) and j < len(right):
+                if left[i][0] <= right[j][0]:
+                    ans[left[i][1]] += smaller
+                    merged.append(left[i])
+                    i += 1
+                else:
+                    smaller += 1
+                    merged.append(right[j])
+                    j += 1
+
+            while i < len(left):
+                ans[left[i][1]] += smaller
+                merged.append(left[i])
+                i += 1
+
+            while j < len(right):
+                merged.append(right[j])
+                j += 1
+
+            return merged
+
+        merge_sort(arr)
+
+        return ans