CSES - Datatähti 2020 alku - Results
Submission details
Task:Merkkijonot
Sender:TrixterTheTux
Submission time:2019-10-10 10:46:43 +0300
Language:PyPy2
Status:READY
Result:46
Feedback
groupverdictscore
#1ACCEPTED15
#2ACCEPTED31
#30
Test results
testverdicttimegroup
#1ACCEPTED0.04 s1, 2, 3details
#2ACCEPTED0.11 s2, 3details
#3--3details
#4ACCEPTED0.09 s3details
#5ACCEPTED0.12 s3details
#6ACCEPTED0.04 s1, 2, 3details
#7ACCEPTED0.04 s1, 2, 3details
#8ACCEPTED0.04 s1, 2, 3details
#9ACCEPTED0.04 s1, 2, 3details
#10ACCEPTED0.10 s2, 3details
#11ACCEPTED0.10 s2, 3details
#12ACCEPTED0.10 s2, 3details
#13ACCEPTED0.10 s2, 3details
#14ACCEPTED0.10 s2, 3details
#15ACCEPTED0.38 s3details
#16ACCEPTED0.39 s3details
#17ACCEPTED0.39 s3details
#18ACCEPTED0.38 s3details
#19ACCEPTED0.38 s3details
#20ACCEPTED0.04 s1, 2, 3details
#21ACCEPTED0.04 s1, 2, 3details
#22ACCEPTED0.04 s1, 2, 3details
#23ACCEPTED0.04 s1, 2, 3details
#24ACCEPTED0.04 s1, 2, 3details

Code

def harmonic(data, index, combinationIndex, stopAt, offset):
    # global ran
    # ran += 1

    # print('process {}, index {}, combination {}, stop at {}, offset {}'.format(data, index, combinationIndex, stopAt, offset))
    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 {}, offset {}'.format(index, stopAt, combinationIndex, offset))
            # print(x, one, zero)
            if offset == stopAt and index + offset == stopAt:
                for str in x:
                    count = counts[str]
                    if count > 1:
                        sorted_len += count - 1

                res += (sorted_len - 1) * sorted_len / 2
            else:
                if combinationIndex == stopAt:
                    res += harmonic(x, 0, offset + 1, stopAt, offset + 1)
                else:
                    res += harmonic(x, index + 1, combinationIndex + 1, stopAt, offset)
        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, 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
 
# ran = 0
print(solve(input))
# print('harmonic func ran {} times'.format(ran))

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:

input
100000
RESKYADLVRSDCDXJVECAJCAP
CZLIWDHKVNRJHZINHFBOQEPN
LSTRSDTLADUIWXFQJSQPVRSPBCTZ
QEMSZOQSSVBRNQTCLJQIGVQO
...

correct output
404880850

user output
(empty)

Test 4

Group: 3

Verdict: ACCEPTED

input
100000
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
4999950000

user output
4999950000

Test 5

Group: 3

Verdict: ACCEPTED

input
100000
IPI
Q
SPGV
UNYE
...

correct output
1041990627

user output
1041990627

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

input
20
A
B
C
D
...

correct output
190

user output
190

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

input
5000
SEKNHNRHSRFYIHSCDPGXPTGUSYBFKZ...

correct output
20861

user output
20861

Test 11

Group: 2, 3

Verdict: ACCEPTED

input
5000
UGPWHWOWAREEIROZTRJGUCKDKTZJEO...

correct output
22818

user output
22818

Test 12

Group: 2, 3

Verdict: ACCEPTED

input
5000
FFQQCBCINBDWIKHFPO
SCJU
DNHYPGIQNZS
KNQCYRAXHHKORXMOVHL
...

correct output
22268

user output
22268

Test 13

Group: 2, 3

Verdict: ACCEPTED

input
5000
QVWMSYPWXWBBXWCVEOUAVQLDWNBXZB...

correct output
22589

user output
22589

Test 14

Group: 2, 3

Verdict: ACCEPTED

input
5000
JYORMNQAKXHNRBWGGSMWXWXMBKPBZU...

correct output
20482

user output
20482

Test 15

Group: 3

Verdict: ACCEPTED

input
100000
DCXXVIEYQAASZJYUUVYZUIJFGVBAYC...

correct output
8925377

user output
8925377

Test 16

Group: 3

Verdict: ACCEPTED

input
100000
IJZHBIYNWQBDWNGZWVXJKXXWFVLPQX...

correct output
8907840

user output
8907840

Test 17

Group: 3

Verdict: ACCEPTED

input
100000
GWUGLGQDAYGGJROIAKBMMICQMDUQQQ...

correct output
8771809

user output
8771809

Test 18

Group: 3

Verdict: ACCEPTED

input
100000
SWVSVXURRSLCSHCPAKWESIPCNDKYLD...

correct output
8704480

user output
8704480

Test 19

Group: 3

Verdict: ACCEPTED

input
100000
ZLSNSZBGIMCOLPDEAKVDAJCTMBNRLH...

correct output
8672362

user output
8672362

Test 20

Group: 1, 2, 3

Verdict: ACCEPTED

input
20
AAB
D
CDBAA
D
...

correct output
15

user output
15

Test 21

Group: 1, 2, 3

Verdict: ACCEPTED

input
20
A
CDD
AABBC
DBCB
...

correct output
15

user output
15

Test 22

Group: 1, 2, 3

Verdict: ACCEPTED

input
20
ACADD
CDDCD
DACA
CCDCA
...

correct output
17

user output
17

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

input
20
DDA
BBCDC
DDD
CCD
...

correct output
14

user output
14