Anagrams

import sys

if __name__ == '__main__':
    n = int(sys.stdin.readline())

    origin_words = list()
    sorted_words = list()
    index = list()

    for i in range(n):
        word = sys.stdin.readline().strip()
        origin_words.append(word)
        sorted_words.append(''.join(sorted(word)))
        index.append(i)

    sorted_sorted_words, index = zip(*sorted(zip(sorted_words, index)))

    anagram_group = dict()
    for i, word in enumerate(sorted_sorted_words):
        if word in anagram_group:
            anagram_group[word].append(origin_words[index[i]])
        else:
            anagram_group[word] = list()
            anagram_group[word].append(origin_words[index[i]])

    print(len(anagram_group.values()))
    for values in anagram_group.values():
        print(len(values))
        for word in values:
            print(word)

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: ACCEPTED

Test 9

Verdict: WRONG ANSWER