乱序字符串

yingl · yingl · commit 8e6c7fed5ca3 · 2016-07-31T20:39:20.000+08:00
diff --git a/anagrams.py b/anagrams.py
@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+
+class Solution:
+    # @param strs: A list of strings
+    # @return: A list of strings
+    def anagrams(self, strs):
+        # write your code here
+        ret = []
+        word_map = {}
+        for word in strs:
+            sorted_word = ''.join(sorted(word))
+            if sorted_word not in word_map:
+                word_map[sorted_word] = {'count':1, 'words':[word]}
+            else:
+                word_map[sorted_word]['count'] += 1
+                word_map[sorted_word]['words'].append(word)
+        for key, info in word_map.items():
+            if info['count'] > 1:
+                for word in info['words']:
+                    ret.append(word)
+        return ret
