adding algo

Author: ivan1016017

src/my_project/interviews/amazon_high_frequency_23/round_2/k_free_subsets_round_1.py

@@ -0,0 +1,46 @@
+from typing import List, Union, Collection, Mapping, Optional
+from collections import defaultdict
+
+class Solution:
+    def countTheNumOfKFreeSubsets(self, nums: List[int], k: int) -> int:
+
+        groups = defaultdict(list)
+
+        for num in nums:
+            groups[num % k].append(num)
+
+        res = 1
+
+        for group in groups.values():
+            group.sort()
+
+            i = 0
+            while i < len(group):
+                chain = [group[i]]
+
+                j = i + 1
+
+                while j < len(group) and group[j] == chain[-1] + k:
+                    chain.append(group[j])
+                    j += 1
+
+                m = len(chain)
+                if m == 1:
+                    chain_res = 2 # {} and {chain[0]}
+                else:
+                    take = 1
+                    skip = 1
+
+                    for i in range(1,m):
+                        new_take = skip 
+                        new_skip = take + skip
+                        take, skip = new_take, new_skip
+
+                    chain_res = take + skip 
+
+
+                res *= chain_res
+
+                i = j 
+
+        return res

src/my_project/interviews/amazon_high_frequency_23/round_2/reorder_data_in_log_file_round_1.py

@@ -0,0 +1,16 @@
+from typing import List, Union, Collection, Mapping, Optional
+
+class Solution:
+    def reorderLogFiles(self, logs: List[str]) -> List[str]:
+
+        letter_logs = []
+        digit_logs = []
+
+        for log in logs:
+            _id, rest = log.split(' ', 1)
+            if rest[0].isdigit():
+                digit_logs.append(log)
+            else:
+                letter_logs.append((rest, _id, log))
+        letter_logs.sort(key=lambda x: (x[0],x[1]))
+        return [orig for _,_,orig in letter_logs] + digit_logs

src/my_project/interviews/amazon_high_frequency_23/round_2/reorder_data_in_log_files.py

@@ -0,0 +1,15 @@
+from typing import List, Union, Collection, Mapping, Optional
+
+class Solution:
+    def reorderLogFiles(self, logs: List[str]) -> List[str]:
+        letter_logs = []
+        digit_logs = []
+        for log in logs:
+            id_, rest = log.split(" ", 1)
+            if rest[0].isdigit():
+                digit_logs.append(log)
+            else:
+                letter_logs.append((rest, id_, log))
+        # sort by content first, then identifier
+        letter_logs.sort(key=lambda x: (x[0], x[1]))
+        return [orig for _, _, orig in letter_logs] + digit_logs