LeetCode-Solutions/Python/group-shifted-strings.py at master · pythonpeixun/LeetCode-Solutions · GitHub

30 lines (23 loc) · 754 Bytes

# Time: O(nlogn)

# Space: O(n)

import collections

class Solution(object):

# @param {string[]} strings

# @return {string[][]}

def groupStrings(self, strings):

groups = collections.defaultdict(list)

for s in strings: # Grouping.

groups[self.hashStr(s)].append(s)

result = []

for key, val in groups.iteritems():

result.append(sorted(val))

return result

def hashStr(self, s):

base = ord(s[0])

hashcode = ""

for i in xrange(len(s)):

if ord(s[i]) - base >= 0:

hashcode += unichr(ord('a') + ord(s[i]) - base)

else:

hashcode += unichr(ord('a') + ord(s[i]) - base + 26)

return hashcode

Expand file tree