Create find-the-maximum-number-of-elements-in-subset.py

Author: kamyu104

Python/find-the-maximum-number-of-elements-in-subset.py

@@ -0,0 +1,26 @@
+# Time:  O(n * log(logr)) = O(9 * n)
+# Space: O(n)
+
+import collections
+
+
+# freq table
+class Solution(object):
+    def maximumLength(self, nums):
+        """
+        :type nums: List[int]
+        :rtype: int
+        """
+        cnt = collections.Counter(nums)
+        result = 0
+        for x in cnt.iterkeys():
+            if x == 1:
+                result = max(result, cnt[x]-(1 if cnt[x]%2 == 0 else 0))
+                continue
+            l = 0
+            while x in cnt and cnt[x] >= 2:
+                l += 2
+                x *= x
+            l += 1 if x in cnt else -1
+            result = max(result, l)
+        return result