CSES - Datatähti 2020 alku

Submission details
Task:	Merkkijonot
Sender:	TrixterTheTux
Submission time:	2019-10-07 20:37:44 +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.04 s	1, 2, 3	details
#2	ACCEPTED	0.11 s	2, 3	details
#3	TIME LIMIT EXCEEDED	--	3	details
#4	ACCEPTED	0.09 s	3	details
#5	WRONG ANSWER	0.12 s	3	details
#6	ACCEPTED	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	WRONG ANSWER	0.04 s	1, 2, 3	details
#10	WRONG ANSWER	0.10 s	2, 3	details
#11	WRONG ANSWER	0.10 s	2, 3	details
#12	WRONG ANSWER	0.09 s	2, 3	details
#13	WRONG ANSWER	0.10 s	2, 3	details
#14	WRONG ANSWER	0.10 s	2, 3	details
#15	WRONG ANSWER	0.41 s	3	details
#16	WRONG ANSWER	0.40 s	3	details
#17	WRONG ANSWER	0.40 s	3	details
#18	WRONG ANSWER	0.40 s	3	details
#19	WRONG ANSWER	0.40 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):
    # print('process {}, index {}, combination {}, stop at {}'.format(data, index, combinationIndex, stopAt))
    sorted = [[], []]
    zero = 0
    one = 0
    for x in data:
        # print(x, x[index], x[combinationIndex])
        optimized = x[index] == x[combinationIndex] and 1 or 0
        sorted[optimized].append(x)

        if optimized:
            one += 1
        else:
            zero += 1
 
    res = 0
    i = 0
    for x in sorted:
        if i == 0:
            sorted_len = zero
            i += 1
        else:
            sorted_len = one

        if sorted_len > 1:
            # print('time to decide actions! based on index {}, stopAt {}, combinationIndex {}'.format(index, stopAt, combinationIndex))
            # print(x, one, zero)
            if index == stopAt and combinationIndex == stopAt:
                res += (sorted_len - 1) * sorted_len / 2

                for str in x:
                    count = counts[str]
                    if count > 1:
                        res += (count - 2) * (count - 1) / 2
            elif combinationIndex < stopAt:
                res += harmonic(x, index, combinationIndex + 1, stopAt)
            else:
                if index + 2 > stopAt:
                    res += (sorted_len - 1) * sorted_len / 2

                    for str in x:
                        count = counts[str]
                        if count > 1:
                            res += (count - 2) * (count - 1) / 2
                else:
                    res += harmonic(x, index + 1, index + 2, stopAt)
        elif sorted_len == 1:
            count = counts[x[0]]
            if count > 1:
                res += (count - 2) * (count - 1) / 2

    return res
 
def solve(input):
    res = 0
 
    for x in input:
        data = input[x]
        data_len = len(data)

        str_len = int(x)
        if str_len > 1:
            res += harmonic(data, 0, 1, str_len - 1)
        else:
            res += (data_len - 1) * data_len / 2

    return res

count = int(raw_input())
input = dict()
counts = dict()
for x in range(0, count):
    curInput = raw_input()
    str_len = len(curInput)
 
    if not str_len in input:
        input[str_len] = list()
        
    if not curInput in counts:
        counts[curInput] = 1
        input[str_len].append(curInput)

    counts[curInput] += 1
 
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: ACCEPTED

input
`100000 AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...` view save

correct output
`4999950000` view save

user output
`4999950000` view save

Test 5

Group: 3

Verdict: WRONG ANSWER

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

correct output
`1041990627` view save

user output
`254530405` view save

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
`190` view save

user output
`190` 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: WRONG ANSWER

input
`2 A A` view save

correct output
`1` view save

user output
`0` view save

Test 10

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 SEKNHNRHSRFYIHSCDPGXPTGUSYBFKZ...` view save

correct output
`20861` view save

user output
`14795` view save

Test 11

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 UGPWHWOWAREEIROZTRJGUCKDKTZJEO...` view save

correct output
`22818` view save

user output
`16671` 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
`16628` view save

Test 13

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 QVWMSYPWXWBBXWCVEOUAVQLDWNBXZB...` view save

correct output
`22589` view save

user output
`17016` view save

Test 14

Group: 2, 3

Verdict: WRONG ANSWER

input
`5000 JYORMNQAKXHNRBWGGSMWXWXMBKPBZU...` view save

correct output
`20482` view save

user output
`15601` view save

Test 15

Group: 3

Verdict: WRONG ANSWER

input
`100000 DCXXVIEYQAASZJYUUVYZUIJFGVBAYC...` view save

correct output
`8925377` view save

user output
`5215684` view save

Test 16

Group: 3

Verdict: WRONG ANSWER

input
`100000 IJZHBIYNWQBDWNGZWVXJKXXWFVLPQX...` view save

correct output
`8907840` view save

user output
`5076953` view save

Test 17

Group: 3

Verdict: WRONG ANSWER

input
`100000 GWUGLGQDAYGGJROIAKBMMICQMDUQQQ...` view save

correct output
`8771809` view save

user output
`4928812` view save

Test 18

Group: 3

Verdict: WRONG ANSWER

input
`100000 SWVSVXURRSLCSHCPAKWESIPCNDKYLD...` view save

correct output
`8704480` view save

user output
`4972166` view save

Test 19

Group: 3

Verdict: WRONG ANSWER

input
`100000 ZLSNSZBGIMCOLPDEAKVDAJCTMBNRLH...` view save

correct output
`8672362` view save

user output
`4870689` 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
`7` 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
`6` 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
`8` 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)