Finish 1452

Author: Khanh Do

1452-people-whose-list-of-favorite-companies-is-not-a-subset-of-another-list.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python
+
+def people_indexes(favorite_companies):
+    """Return the indices of people whose list of favorite companies is not a
+    subset of any other list of favorite companies.
+
+    Time: O(n^2) -- nested loops to compare current list with all other lists
+    Space: O(n) -- all lists may be disjoint with one another
+    """
+    elim_indexes = set()
+
+    for i, u in enumerate(favorite_companies):
+        for j, v in enumerate(favorite_companies):
+            if j != i:  # Exclude self-comparison
+                if set(u).issubset(set(v)):
+                    elim_indexes.add(i)
+
+    all_indexes = set(range(len(favorite_companies)))
+    return list(all_indexes - elim_indexes)
+
+
+def main():
+    example_cases = [
+        (
+            [
+                ['leetcode', 'google', 'facebook'],
+                ['google', 'microsoft'],
+                ['google', 'facebook'],
+                ['google'],
+                ['amazon'],
+            ],
+            [0, 1, 4]),
+        (
+            [
+                ['leetcode', 'google', 'facebook'],
+                ['leetcode', 'amazon'],
+                ['facebook', 'google'],
+            ],
+            [0, 1]
+        ),
+        (
+            [
+                ['leetcode'],
+                ['google'],
+                ['facebook'],
+                ['amazon'],
+            ],
+            [0, 1, 2, 3]
+        ),
+
+    ]
+
+    for favorite_companies, ans in example_cases:
+        assert people_indexes(favorite_companies) == ans
+
+
+    corner_cases = [
+        (
+            [],            # len(favorite_companies) == 0
+            []
+        ),
+        (
+            [['google']],  # len(favorite_companies) == 1
+            [0]
+        ),
+    ]
+
+    for favorite_companies, ans in corner_cases:
+        assert people_indexes(favorite_companies) == ans
+
+
+if __name__ == '__main__':
+    main()