Merkkijonot

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

parser = argparse.ArgumentParser()
parser.add_argument("-d", "--debug", help="If you want extra information then pass this argument into the program", action="store_true")
parser.add_argument("-t", "--test", help="If you want to generate random strings for testing", action="store")
args = parser.parse_args()
debug = args.debug
testOrNo = args.test

# ESIM AAB JA SSG OVAT HARMOONINEN MERKKIJONOPARI
merkkijonot = []
harmoonisten_merkkijonoparien_maara = 0
alreadyFoundJonot = []
threadsSpawned = 0

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

if testOrNo:
	for i in range(int(args.test)):
	 	merkkijonot.append(str(randomString()))

else:
	spawnattavien_merkkijonojen_maara = int(input())
	for i in range(spawnattavien_merkkijonojen_maara):
 		merkkijonot.append(str(input()))
# 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)

def check(start, step):
	for a in range(start,len(merkkijonot)):
		showed = False
		for b in range(start,len(merkkijonot), step):
			# if start == 0: print("t1:", a, b)
			# if start == 1: print("	t2:", a, b)
			# 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: ")
				global harmoonisten_merkkijonoparien_maara
				harmoonisten_merkkijonoparien_maara += 1
				if debug: print("true")
				continue
			if debug: print("	false")
	global threadsSpawned
	threadsSpawned -= 1

if debug: print("ilman pyöristystä: ", harmoonisten_merkkijonoparien_maara)

# 0 3 6 : 0, 3
# 1 4 7 : 1, 3
# 2,5,8 : 2, 3

# threads_to_spawn = n
# check(n-i, n)
threads_to_spawn = 2000
threadsSpawned = threads_to_spawn
# for i in range(threads_to_spawn):
# 	newThread = threading.Thread(target = check, args = (i, threads_to_spawn))
# 	newThread.start()
	
for i in range(threads_to_spawn):
	newThread = threading.Thread(target = check, args = (i, threads_to_spawn))
	newThread.start()

while True:
	if threadsSpawned == 0:
		break
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