CSES - Datatähti 2020 alku

Submission details
Task:	Merkkijonot
Sender:	TrixterTheTux
Submission time:	2019-10-07 15:09:52 +0300
Language:	Python2 (PyPy2)
Status:	READY
Result:	0

Feedback
group	verdict	score
#1	WRONG ANSWER	0
#2	WRONG ANSWER	0
#3	TIME LIMIT EXCEEDED	0

Test results
test	verdict	time	group
#1	ACCEPTED	0.05 s	1, 2, 3	details
#2	ACCEPTED	0.13 s	2, 3	details
#3	TIME LIMIT EXCEEDED	--	3	details
#4	TIME LIMIT EXCEEDED	--	3	details
#5	WRONG ANSWER	0.14 s	3	details
#6	WRONG ANSWER	0.04 s	1, 2, 3	details
#7	ACCEPTED	0.04 s	1, 2, 3	details
#8	ACCEPTED	0.04 s	1, 2, 3	details
#9	ACCEPTED	0.04 s	1, 2, 3	details
#10	WRONG ANSWER	0.09 s	2, 3	details
#11	WRONG ANSWER	0.09 s	2, 3	details
#12	WRONG ANSWER	0.09 s	2, 3	details
#13	WRONG ANSWER	0.09 s	2, 3	details
#14	WRONG ANSWER	0.09 s	2, 3	details
#15	WRONG ANSWER	0.35 s	3	details
#16	WRONG ANSWER	0.34 s	3	details
#17	WRONG ANSWER	0.35 s	3	details
#18	WRONG ANSWER	0.34 s	3	details
#19	WRONG ANSWER	0.34 s	3	details
#20	WRONG ANSWER	0.04 s	1, 2, 3	details
#21	WRONG ANSWER	0.04 s	1, 2, 3	details
#22	WRONG ANSWER	0.04 s	1, 2, 3	details
#23	ACCEPTED	0.04 s	1, 2, 3	details
#24	WRONG ANSWER	0.04 s	1, 2, 3	details

Code

def harmonic(data, index, combinationIndex, stopAt):
    sorted = [[], []]
    for x in data:
        optimized = x[index] == x[combinationIndex] and 1 or 0
        # print('COMPARE FOR {}: {}, {} ({}, {})'.format(x, x[index], x[combinationIndex], index, combinationIndex))
 
        sorted[optimized].append(x)
    
    res = 0
    for data in sorted:
        sorted_len = len(data)
        if sorted_len > 1:
            if index < stopAt:
                res += harmonic(data, index + 1, combinationIndex, stopAt)
            elif combinationIndex < stopAt:
                res += harmonic(data, 0, combinationIndex + 1, stopAt)
            else:
                res += (sorted_len - 1) * sorted_len / 2

    return res
 
def solve(input):
    res = 0
 
    for index in input:
        data = input[index]
        if len(data) < 2:
            continue

        str_len = int(index)
        if str_len > 1:
            res += harmonic(data, 0, 1, str_len - 1)
        else:
            for x in list(set(data)):
                count = data.count(x)
                if count > 1:
                    res += (count - 1) * count / 2
 
    return res
 
count = int(raw_input())
input = dict()
for x in range(0, count):
    curInput = raw_input()
    str_len = str(len(curInput))
    if not str_len in input:
        input[str_len] = list()
 
    input[str_len].append(curInput)
 
print(solve(input))

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
`20 FFDISOVJLPIASANJRBHQXJVW HBGVRZFWOTAAVUPYKMKXAOBEYUE JKHXTPHUKTLTLNXMXMSYYFLF HCSZMXTZCUFHIHSTEWKHCFZRTXU ...` view save

correct output
`20` view save

user output
`20` view save

Test 2

Group: 2, 3

Verdict: ACCEPTED

input
`5000 AJEDROEDXRGDUSDAPZBXXQBPJ PDGCTAGCRTBCJICPFELRRMLFD TISAOTGMDKUCITIEFQVCSBZJYUUO FMASVJYFPMELSVLQWFZVLVMSPHYL ...` view save

correct output
`2178556` view save

user output
`2178556` view save

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED

input
`100000 RESKYADLVRSDCDXJVECAJCAP CZLIWDHKVNRJHZINHFBOQEPN LSTRSDTLADUIWXFQJSQPVRSPBCTZ QEMSZOQSSVBRNQTCLJQIGVQO ...` view save

correct output
`404880850` view save

user output
(empty)

Test 4

Group: 3

Verdict: TIME LIMIT EXCEEDED

input
`100000 AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...` view save

correct output
`4999950000` view save

user output
(empty)

Test 5

Group: 3

Verdict: WRONG ANSWER

input
`100000 IPI Q SPGV UNYE ...` view save

correct output
`1041990627` view save

user output
`739718318` view save

Test 6

Group: 1, 2, 3

Verdict: WRONG ANSWER

input
`20 A B C D ...` view save

correct output
`190` view save

user output
`0` view save

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

input
`1 A` view save

correct output
`0` view save

user output
`0` view save

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

input
`20 A BB CCC DDDD ...` view save

correct output
`0` view save

user output
`0` view save

Test 9

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2 A A` view save

correct output
`1` view save

user output
`1` view save

Test 10

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 SEKNHNRHSRFYIHSCDPGXPTGUSYBFKZ...` view save

correct output
`20861` view save

user output
`14857` view save

Test 11

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 UGPWHWOWAREEIROZTRJGUCKDKTZJEO...` view save

correct output
`22818` view save

user output
`17163` view save

Test 12

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 FFQQCBCINBDWIKHFPO SCJU DNHYPGIQNZS KNQCYRAXHHKORXMOVHL ...` view save

correct output
`22268` view save

user output
`16999` view save

Test 13

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 QVWMSYPWXWBBXWCVEOUAVQLDWNBXZB...` view save

correct output
`22589` view save

user output
`17561` view save

Test 14

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 JYORMNQAKXHNRBWGGSMWXWXMBKPBZU...` view save

correct output
`20482` view save

user output
`16277` view save

Test 15

Group: 3

Verdict: WRONG ANSWER

input
`100000 DCXXVIEYQAASZJYUUVYZUIJFGVBAYC...` view save

correct output
`8925377` view save

user output
`7132154` view save

Test 16

Group: 3

Verdict: WRONG ANSWER

input
`100000 IJZHBIYNWQBDWNGZWVXJKXXWFVLPQX...` view save

correct output
`8907840` view save

user output
`6964620` view save

Test 17

Group: 3

Verdict: WRONG ANSWER

input
`100000 GWUGLGQDAYGGJROIAKBMMICQMDUQQQ...` view save

correct output
`8771809` view save

user output
`6795545` view save

Test 18

Group: 3

Verdict: WRONG ANSWER

input
`100000 SWVSVXURRSLCSHCPAKWESIPCNDKYLD...` view save

correct output
`8704480` view save

user output
`6879308` view save

Test 19

Group: 3

Verdict: WRONG ANSWER

input
`100000 ZLSNSZBGIMCOLPDEAKVDAJCTMBNRLH...` view save

correct output
`8672362` view save

user output
`6831922` view save

Test 20

Group: 1, 2, 3

Verdict: WRONG ANSWER

input
`20 AAB D CDBAA D ...` view save

correct output
`15` view save

user output
`8` view save

Test 21

Group: 1, 2, 3

Verdict: WRONG ANSWER

input
`20 A CDD AABBC DBCB ...` view save

correct output
`15` view save

user output
`9` view save

Test 22

Group: 1, 2, 3

Verdict: WRONG ANSWER

input
`20 ACADD CDDCD DACA CCDCA ...` view save

correct output
`17` view save

user output
`4` view save

Test 23

Group: 1, 2, 3

Verdict: ACCEPTED

input
`20 DCDA BDBA CACBD BA ...` view save

correct output
`7` view save

user output
`7` view save

Test 24

Group: 1, 2, 3

Verdict: WRONG ANSWER

input
`20 DDA BBCDC DDD CCD ...` view save

correct output
`14` view save

user output
`9` view save

Datatähti 2020 alku

Merkkijonot

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Tasks

Submissions (TrixterTheTux)