Update guess-the-word.py

Author: kamyu104

Python/guess-the-word.py

@@ -1,11 +1,29 @@
-# Time:  O(n^2)
+# Time:  O(n)
 # Space: O(n)
 
 import collections
 import itertools
 
 
 class Solution(object):
+    def findSecretWord(self, wordlist, master):
+        """
+        :type wordlist: List[Str]
+        :type master: Master
+        :rtype: None
+        """
+        possible = range(len(wordlist))
+        n = 0
+        while n < 6:
+            count = [collections.Counter(w[i] for w in wordlist) for i in xrange(6)]
+            guess = max(possible, key=lambda x: sum(count[i][c] for i, c in enumerate(wordlist[x])))
+            n = master.guess(wordlist[guess])
+            possible = [j for j in possible if sum(a == b for a, b in itertools.izip(wordlist[guess], wordlist[j])) == n]
+
+
+# Time:  O(n^2)
+# Space: O(n)
+class Solution2(object):
     def findSecretWord(self, wordlist, master):
         """
         :type wordlist: List[Str]
@@ -37,7 +55,7 @@ def solve(H, possible):
 
 # Time:  O(n^2)
 # Space: O(n)
-class Solution2(object):
+class Solution3(object):
     def findSecretWord(self, wordlist, master):
         """
         :type wordlist: List[Str]