Solução Codeforce 4C em Python

n = int(input())
d = {}
for i in range(0, n):
    s = input()
    if s in d:
        print(s+str(d[s]))
        d[s] += 1
    else:
        print("OK")
        d[s] = 1
Concerned Cottonmouth