CSES - Datatähti 2020 alku

CSES - Datatähti 2020 alku - Results

Submission details
Task:	Merkkijonot
Sender:	TrixterTheTux
Submission time:	2019-10-07 21:18:15 +0300
Language:	Python2 (PyPy2)
Status:	READY
Result:	46

Feedback
group	verdict	score
#1	ACCEPTED	15
#2	ACCEPTED	31
#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	ACCEPTED	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	ACCEPTED	0.04 s	1, 2, 3	details
#10	ACCEPTED	0.10 s	2, 3	details
#11	ACCEPTED	0.09 s	2, 3	details
#12	ACCEPTED	0.09 s	2, 3	details
#13	ACCEPTED	0.09 s	2, 3	details
#14	ACCEPTED	0.09 s	2, 3	details
#15	ACCEPTED	0.38 s	3	details
#16	ACCEPTED	0.38 s	3	details
#17	ACCEPTED	0.39 s	3	details
#18	ACCEPTED	0.38 s	3	details
#19	ACCEPTED	0.38 s	3	details
#20	ACCEPTED	0.04 s	1, 2, 3	details
#21	ACCEPTED	0.04 s	1, 2, 3	details
#22	ACCEPTED	0.04 s	1, 2, 3	details
#23	ACCEPTED	0.04 s	1, 2, 3	details
#24	ACCEPTED	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 combinationIndex < stopAt:
                res += harmonic(x, index, combinationIndex + 1, stopAt)
            else:
                if index + 2 > stopAt:
                    for str in x:
                        count = counts[str]
                        if count > 1:
                            sorted_len += count - 1

                    res += (sorted_len - 1) * sorted_len / 2
                else:
                    res += harmonic(x, index + 1, index + 2, stopAt)
        elif sorted_len == 1:
            count = counts[x[0]]
            if count > 1:
                res += (count - 1) * count / 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:
            for str in data:
                count = counts[str]
                if count > 1:
                    data_len += count - 1

            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)
    else:
        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: ACCEPTED

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

correct output
`1041990627` view save

user output
`1041990627` 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: ACCEPTED

input
`2 A A` view save

correct output
`1` view save

user output
`1` view save

Test 10

Group: 2, 3

Verdict: ACCEPTED

input
`5000 SEKNHNRHSRFYIHSCDPGXPTGUSYBFKZ...` view save

correct output
`20861` view save

user output
`20861` view save

Test 11

Group: 2, 3

Verdict: ACCEPTED

input
`5000 UGPWHWOWAREEIROZTRJGUCKDKTZJEO...` view save

correct output
`22818` view save

user output
`22818` view save

Test 12

Group: 2, 3

Verdict: ACCEPTED

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

correct output
`22268` view save

user output
`22268` view save

Test 13

Group: 2, 3

Verdict: ACCEPTED

input
`5000 QVWMSYPWXWBBXWCVEOUAVQLDWNBXZB...` view save

correct output
`22589` view save

user output
`22589` view save

Test 14

Group: 2, 3

Verdict: ACCEPTED

input
`5000 JYORMNQAKXHNRBWGGSMWXWXMBKPBZU...` view save

correct output
`20482` view save

user output
`20482` view save

Test 15

Group: 3

Verdict: ACCEPTED

input
`100000 DCXXVIEYQAASZJYUUVYZUIJFGVBAYC...` view save

correct output
`8925377` view save

user output
`8925377` view save

Test 16

Group: 3

Verdict: ACCEPTED

input
`100000 IJZHBIYNWQBDWNGZWVXJKXXWFVLPQX...` view save

correct output
`8907840` view save

user output
`8907840` view save

Test 17

Group: 3

Verdict: ACCEPTED

input
`100000 GWUGLGQDAYGGJROIAKBMMICQMDUQQQ...` view save

correct output
`8771809` view save

user output
`8771809` view save

Test 18

Group: 3

Verdict: ACCEPTED

input
`100000 SWVSVXURRSLCSHCPAKWESIPCNDKYLD...` view save

correct output
`8704480` view save

user output
`8704480` view save

Test 19

Group: 3

Verdict: ACCEPTED

input
`100000 ZLSNSZBGIMCOLPDEAKVDAJCTMBNRLH...` view save

correct output
`8672362` view save

user output
`8672362` view save

Test 20

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
`15` view save

user output
`15` view save

Test 21

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
`15` view save

user output
`15` view save

Test 22

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
`17` view save

user output
`17` 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: ACCEPTED

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

correct output
`14` view save

user output
`14` 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)