Anagrams

def main() :
    n = int(input())
    arr = [] 
    for _ in range(n) :
        arr.append(input())
    
    check = {}
    anagrams = []
    for i in range(len(arr)) :
        sorted_string = "".join(sorted(arr[i]))
        if sorted_string in anagrams : 
            (pos, count) = check[sorted_string]
            pos.append(i)
            check[sorted_string] = [pos,count+1]
        else :
            anagrams.append(sorted_string)
            check[sorted_string] = [[i],1]
    anagrams_count = 0

    for i in check :
        if check[i][1] > 1 :
            anagrams_count += 1


    print(anagrams_count)
    for i in check : 
        if check[i][1] > 1 :
            print(check[i][1])
            for j in check[i][0] :
                print(arr[j])


if __name__ == "__main__":
    main()

Test 1

Verdict: TIME LIMIT EXCEEDED

Test 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Verdict: TIME LIMIT EXCEEDED

Test 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED