CSES - Datatähti 2024 alku - Results
Submission details
Task:Uolevin kalansaalis
Sender:d-m-g
Submission time:2023-11-09 23:35:56 +0200
Language:Python3 (CPython3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1, 2details
#20.02 s1, 2details
#3ACCEPTED0.02 s1, 2details
#4ACCEPTED0.02 s1, 2details
#5ACCEPTED0.02 s1, 2details
#60.02 s1, 2details
#7ACCEPTED0.02 s1, 2details
#80.02 s1, 2details
#9ACCEPTED0.02 s1, 2details
#10ACCEPTED0.02 s1, 2details
#11ACCEPTED0.02 s1, 2details
#12ACCEPTED0.02 s1, 2details
#13ACCEPTED0.02 s1, 2details
#140.02 s1, 2details
#150.02 s1, 2details
#16--2details
#17--2details
#18--2details
#19--2details
#20--2details
#21--2details
#22--2details
#23--2details
#24--2details

Code

def main():
    n, m, k = [int(el) for el in input().split()]
    pikes = set()
    shrimps = set()
    whole_sum = 0
    for i in range(k):
        x, y, t = [el for el in input().split()]
        x = int(x)
        y = int(y)
        if t == "K":
            shrimps.add((x - 1, y - 1))
            whole_sum -= 10
        else:
            pikes.add((x - 1, y - 1))
            whole_sum += 1

    res = -10 * k
    for i in range(n):
        
        for j in range(m):
            for size in range(1, min(m, n) + 1):
                if i == 4 and j == 1 and size == 4:
                    pass
                for rotation in range(2):
                    if (rotation == 0 and m - j >= size and n - i >= size) or (rotation == 1 and m - j >= size and i >= size):
                        res = max(get_cost_of_a_triangle(i, j, size, rotation, pikes, shrimps, whole_sum), res)

    print(res)

def get_cost_of_a_triangle(i, j, size, rotation, pikes, shrimps, whole_sum):
    cost = 0
    if rotation == 0:
        y = j
        for x in range(i, i + size):
            end_of_the_str = j + size - (x - i) // 2
            while y < end_of_the_str:
                if (x, y) in pikes:
                    cost += 1
                elif (x, y) in shrimps:
                    cost -= 10
    
                y += 1
            y = j + (x - i + 2) // 2

    else:
        y = j
        r = [i - el for el in range(size)]
        #print(r)
        for x in r:
            end_of_the_str = j + size - (i - x) // 2
            while y < end_of_the_str:
                if (x, y) in pikes:
                    cost += 1
                elif (x, y) in shrimps:
                    cost -= 10
    
                y += 1
            y = j + (i - x + 1) // 2
        #print(f"coordinates: {i}, {j}; size: {size} ---> {cost}")
    return whole_sum - cost
main()

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
5 6 13
1 1 K
5 1 K
2 2 H
4 2 H
...

correct output
-16

user output
-16

Test 2

Group: 1, 2

Verdict:

input
5 6 7
1 5 K
4 6 K
2 4 H
2 5 H
...

correct output
0

user output
-5

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
5 6 7
5 5 K
2 6 K
2 4 H
2 5 H
...

correct output
0

user output
0

Test 4

Group: 1, 2

Verdict: ACCEPTED

input
10 10 51
3 3 H
6 3 H
9 5 H
5 10 H
...

correct output
50

user output
50

Test 5

Group: 1, 2

Verdict: ACCEPTED

input
10 10 52
3 5 H
3 1 H
9 6 H
2 8 H
...

correct output
40

user output
40

Test 6

Group: 1, 2

Verdict:

input
10 10 60
6 10 H
2 8 H
5 8 H
8 10 H
...

correct output
-15

user output
-18

Test 7

Group: 1, 2

Verdict: ACCEPTED

input
10 10 60
4 7 H
7 4 H
4 10 H
3 6 H
...

correct output
60

user output
60

Test 8

Group: 1, 2

Verdict:

input
10 10 40
9 9 H
5 10 H
5 6 H
4 9 H
...

correct output
2

user output
12

Test 9

Group: 1, 2

Verdict: ACCEPTED

input
1 1 0

correct output
0

user output
0

Test 10

Group: 1, 2

Verdict: ACCEPTED

input
1 1 1
1 1 K

correct output
0

user output
0

Test 11

Group: 1, 2

Verdict: ACCEPTED

input
1 1 1
1 1 H

correct output
0

user output
0

Test 12

Group: 1, 2

Verdict: ACCEPTED

input
10 5 32
10 3 H
4 4 H
3 3 H
5 4 H
...

correct output
20

user output
20

Test 13

Group: 1, 2

Verdict: ACCEPTED

input
5 10 32
5 9 H
2 4 H
2 9 H
2 5 H
...

correct output
28

user output
28

Test 14

Group: 1, 2

Verdict:

input
10 10 100
2 9 H
5 4 H
5 9 K
6 1 K
...

correct output
-439

user output
-450

Test 15

Group: 1, 2

Verdict:

input
10 10 100
8 9 H
5 10 H
5 4 H
3 9 H
...

correct output
88

user output
90

Test 16

Group: 2

Verdict:

input
500 500 125000
125 261 K
84 78 K
11 200 K
481 246 K
...

correct output
-624270

user output
(empty)

Test 17

Group: 2

Verdict:

input
500 500 125100
16 61 H
37 62 H
459 125 H
318 476 H
...

correct output
124020

user output
(empty)

Test 18

Group: 2

Verdict:

input
500 500 249999
22 214 H
356 145 H
341 29 H
393 262 H
...

correct output
249999

user output
(empty)

Test 19

Group: 2

Verdict:

input
500 500 32000
30 81 H
315 34 H
78 112 H
367 166 H
...

correct output
10126

user output
(empty)

Test 20

Group: 2

Verdict:

input
500 500 126745
164 390 H
126 331 H
164 126 H
55 92 H
...

correct output
-104692

user output
(empty)

Test 21

Group: 2

Verdict:

input
500 500 71200
106 191 H
314 189 H
482 485 H
344 401 H
...

correct output
-335853

user output
(empty)

Test 22

Group: 2

Verdict:

input
500 500 67772
421 277 H
428 470 H
169 142 H
256 345 H
...

correct output
-208567

user output
(empty)

Test 23

Group: 2

Verdict:

input
500 500 27434
366 481 H
38 22 H
126 107 H
135 169 H
...

correct output
-57100

user output
(empty)

Test 24

Group: 2

Verdict:

input
500 500 93982
183 13 H
463 230 H
264 351 H
399 290 H
...

correct output
-52800

user output
(empty)