[039] Simply check_word with set comprehension

pickfire · web-flow · commit 98c41895cc86 · 2017-11-30T23:46:21.000+08:00
diff --git a/039/word_checker.py b/039/word_checker.py
@@ -12,16 +12,8 @@ def load_dictionary():
     return words
 	
 def check_word(dictionary, word):
-    valid_words = set()
-    permutations = []
-    perms = list(itertools.permutations(word))
-    for i in perms:
-        permutations.append(''.join(i))
-
-    for i in permutations:
-        if i in dictionary:
-            valid_words.add(i)
-    return valid_words
+    permutations = set(map(''.join, itertools.permutations(word)))
+    return {w for w in permutations if w in dictionary}
 	
 def print_words(words):
     for i in words:
