[Leetcode][Python]49: Anagrams

最新推荐文章于 2024-03-27 18:39:09 发布

转载最新推荐文章于 2024-03-27 18:39:09 发布 · 134 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Dabay/p/4367581.html

文章标签：

#python

本文详细解析了LeetCode第49题“字母异位词”的解决方案，通过使用哈希表对字符串进行排序并分组，实现了对一组字符串中所有字母异位词的有效识别与归类。

# -*- coding: utf8 -*-
'''
__author__ = 'dabay.wang@gmail.com'

49: Anagrams
https://leetcode.com/problems/anagrams/

Given an array of strings, return all groups of strings that are anagrams.
Note: All inputs will be in lower-case.

=== Comments by Dabay===
http://blog.youkuaiyun.com/linhuanmars/article/details/21664747
'''

class Solution:
    # @param strs, a list of strings
    # @return a list of strings
    def anagrams(self, strs):
        hash_table = {}
        for str in strs:
            l = list(str)
            l.sort()
            sorted_str = ''.join(l)
            if sorted_str in hash_table:
                hash_table[sorted_str].append(str)
            else:
                hash_table[sorted_str] = [str]
        res = []
        for sorted_str in hash_table:
            if len(hash_table[sorted_str]) > 1:
                res.extend(hash_table[sorted_str])
        return res


def main():
    sol = Solution()
    strs = ["ab", "ba", "abc", "cba", "z"]
    print sol.anagrams(strs)


if __name__ == "__main__":
    import time
    start = time.clock()
    main()
    print "%s sec" % (time.clock() - start)