Merkkijonot

# Tuloste:
# 4
import argparse
import random
import string
import time

parser = argparse.ArgumentParser()
parser.add_argument("-d", "--debug", help="If you want extra information then pass this argument into the program", action="store_true")
args = parser.parse_args()
debug = args.debug

# ESIM AAB JA SSG OVAT HARMOONINEN MERKKIJONOPARI
merkkijonot = []

spawnattavien_merkkijonojen_maara = int(input())

harmoonisten_merkkijonoparien_maara = 0

for i in range(spawnattavien_merkkijonojen_maara):
 	merkkijonot.append(str(input()))
alreadyFoundJonot = []

def randomString():
	stringLength = random.randrange(1,51)
	letters = string.ascii_lowercase
	return (''.join(random.choice(letters) for i in range(stringLength)))

# for i in range():
# 	print(i, "adding")
# 	merkkijonot.append(str(randomString()))


# SSG  SSG
# AAAB AAAB
# SSSS SSSS
# SDSS SDSS

start_time = time.time()
time.time

pari_table_table = []

for x in range(len(merkkijonot)):
	pari_table_table.append([])
	pari_table = [] # pari on 1 ja epäpari on 0
	for i in range(len(merkkijonot[x])):
		myTempArr = []
		for j in range(len(merkkijonot[x])):
			if merkkijonot[x][i] == merkkijonot[x][j]:
				myTempArr.append(1)
			elif merkkijonot[x][i] != merkkijonot[x][j]:
				myTempArr.append(0)

		pari_table_table[x].append(myTempArr)

for a in range(len(merkkijonot)):
	showed = False
	for b in range(len(merkkijonot)):
		# tarkista etä ovatko merkkijono1 ja merkkijonot2 harmooninen merkkijonopari
		merkkijono1 = merkkijonot[a]
		merkkijono2 = merkkijonot[b]

		#print("iterating: ", merkkijono1, merkkijono2)
		if len(merkkijono1) != len(merkkijono2) or b == a:
			#print(merkkijono1, "and", merkkijono2, "are not same length, aborting")
			continue

		if debug: print("	comparing:", f"({merkkijono1})", f"({merkkijono2})")
		
		if debug: print("comparing", pari_table_table[a], " and ", pari_table_table[b])
		if pari_table_table[a] == pari_table_table[b] and (((b,a)) not in alreadyFoundJonot): #(tuple((merkkijono1, merkkijono2)) not in alreadyFoundJonot and tuple((merkkijono2, merkkijono1)) not in alreadyFoundJonot):
			# alreadyFoundJonot.append((merkkijono1, merkkijono2))
			alreadyFoundJonot.append((a,b))
			if debug: print("    paritable found: ")
			harmoonisten_merkkijonoparien_maara += 1
			if debug: print("true")
			continue
		if debug: print("	false")

if debug: print("ilman pyöristystä: ", harmoonisten_merkkijonoparien_maara)
print(harmoonisten_merkkijonoparien_maara)
#print("My program took", time.time() - start_time, "to run")

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

Test 9

Group: 1, 2, 3

Verdict: ACCEPTED

Test 10

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 18

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 19

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 1, 2, 3

Verdict: ACCEPTED

Test 21

Group: 1, 2, 3

Verdict: ACCEPTED

Test 22

Group: 1, 2, 3

Verdict: ACCEPTED

Test 23

Group: 1, 2, 3

Verdict: ACCEPTED

Test 24

Group: 1, 2, 3

Verdict: ACCEPTED