CSES - Datatähti 2020 alku - Results
Submission details
Task:Merkkijonot
Sender:TrixterTheTux
Submission time:2019-10-07 15:17:54 +0300
Language:PyPy2
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.04 s1, 2, 3details
#2ACCEPTED0.10 s2, 3details
#3ACCEPTED0.95 s3details
#4--3details
#50.13 s3details
#60.04 s1, 2, 3details
#7ACCEPTED0.04 s1, 2, 3details
#8ACCEPTED0.04 s1, 2, 3details
#9ACCEPTED0.04 s1, 2, 3details
#100.09 s2, 3details
#110.08 s2, 3details
#120.08 s2, 3details
#130.09 s2, 3details
#140.09 s2, 3details
#150.29 s3details
#160.29 s3details
#170.30 s3details
#180.30 s3details
#190.29 s3details
#200.04 s1, 2, 3details
#210.04 s1, 2, 3details
#220.04 s1, 2, 3details
#23ACCEPTED0.04 s1, 2, 3details
#240.04 s1, 2, 3details

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 combinationIndex < stopAt:
                res += harmonic(data, index, combinationIndex + 1, stopAt)
            elif index < stopAt:
                if index + 2 <= stopAt:
                    res += harmonic(data, index + 1, index + 2, stopAt)
                else:
                    res += (sorted_len - 1) * sorted_len / 2
            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
...

correct output
20

user output
20

Test 2

Group: 2, 3

Verdict: ACCEPTED

input
5000
AJEDROEDXRGDUSDAPZBXXQBPJ
PDGCTAGCRTBCJICPFELRRMLFD
TISAOTGMDKUCITIEFQVCSBZJYUUO
FMASVJYFPMELSVLQWFZVLVMSPHYL
...

correct output
2178556

user output
2178556

Test 3

Group: 3

Verdict: ACCEPTED

input
100000
RESKYADLVRSDCDXJVECAJCAP
CZLIWDHKVNRJHZINHFBOQEPN
LSTRSDTLADUIWXFQJSQPVRSPBCTZ
QEMSZOQSSVBRNQTCLJQIGVQO
...

correct output
404880850

user output
404880850

Test 4

Group: 3

Verdict:

input
100000
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
4999950000

user output
(empty)

Test 5

Group: 3

Verdict:

input
100000
IPI
Q
SPGV
UNYE
...

correct output
1041990627

user output
739718318

Test 6

Group: 1, 2, 3

Verdict:

input
20
A
B
C
D
...

correct output
190

user output
0

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
A

correct output
0

user output
0

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

input
20
A
BB
CCC
DDDD
...

correct output
0

user output
0

Test 9

Group: 1, 2, 3

Verdict: ACCEPTED

input
2
A
A

correct output
1

user output
1

Test 10

Group: 2, 3

Verdict:

input
5000
SEKNHNRHSRFYIHSCDPGXPTGUSYBFKZ...

correct output
20861

user output
14857

Test 11

Group: 2, 3

Verdict:

input
5000
UGPWHWOWAREEIROZTRJGUCKDKTZJEO...

correct output
22818

user output
17163

Test 12

Group: 2, 3

Verdict:

input
5000
FFQQCBCINBDWIKHFPO
SCJU
DNHYPGIQNZS
KNQCYRAXHHKORXMOVHL
...

correct output
22268

user output
16999

Test 13

Group: 2, 3

Verdict:

input
5000
QVWMSYPWXWBBXWCVEOUAVQLDWNBXZB...

correct output
22589

user output
17561

Test 14

Group: 2, 3

Verdict:

input
5000
JYORMNQAKXHNRBWGGSMWXWXMBKPBZU...

correct output
20482

user output
16277

Test 15

Group: 3

Verdict:

input
100000
DCXXVIEYQAASZJYUUVYZUIJFGVBAYC...

correct output
8925377

user output
7132154

Test 16

Group: 3

Verdict:

input
100000
IJZHBIYNWQBDWNGZWVXJKXXWFVLPQX...

correct output
8907840

user output
6964620

Test 17

Group: 3

Verdict:

input
100000
GWUGLGQDAYGGJROIAKBMMICQMDUQQQ...

correct output
8771809

user output
6795545

Test 18

Group: 3

Verdict:

input
100000
SWVSVXURRSLCSHCPAKWESIPCNDKYLD...

correct output
8704480

user output
6879308

Test 19

Group: 3

Verdict:

input
100000
ZLSNSZBGIMCOLPDEAKVDAJCTMBNRLH...

correct output
8672362

user output
6831922

Test 20

Group: 1, 2, 3

Verdict:

input
20
AAB
D
CDBAA
D
...

correct output
15

user output
8

Test 21

Group: 1, 2, 3

Verdict:

input
20
A
CDD
AABBC
DBCB
...

correct output
15

user output
9

Test 22

Group: 1, 2, 3

Verdict:

input
20
ACADD
CDDCD
DACA
CCDCA
...

correct output
17

user output
4

Test 23

Group: 1, 2, 3

Verdict: ACCEPTED

input
20
DCDA
BDBA
CACBD
BA
...

correct output
7

user output
7

Test 24

Group: 1, 2, 3

Verdict:

input
20
DDA
BBCDC
DDD
CCD
...

correct output
14

user output
9