CSES - Datatähti 2024 alku - Results
Submission details
Task:Käännöt
Sender:Lelleri
Submission time:2023-11-12 15:09:49 +0200
Language:CPython3
Status:READY
Result:13
Feedback
groupverdictscore
#1ACCEPTED13
#20
#30
#40
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1details
#2ACCEPTED0.02 s1details
#3ACCEPTED0.02 s1details
#4ACCEPTED0.35 s1details
#5ACCEPTED0.78 s1details
#6ACCEPTED0.02 s2details
#7ACCEPTED0.02 s2details
#8ACCEPTED0.02 s2details
#9ACCEPTED0.02 s2details
#10ACCEPTED0.02 s2details
#11--2details
#12--2details
#13--2details
#14--2details
#15ACCEPTED0.22 s2details
#16--2details
#17ACCEPTED0.02 s3details
#180.02 s3details
#190.11 s3details
#200.07 s3details
#210.04 s3details
#220.15 s3details
#230.09 s3details
#24ACCEPTED0.04 s3details
#250.02 s3details
#26ACCEPTED0.02 s4details
#27ACCEPTED0.02 s4details
#280.05 s4details
#290.02 s4details
#300.07 s4details
#310.09 s4details
#32ACCEPTED0.16 s4details
#33--4details
#340.02 s4details

Code

import time 
 
def sum_lists(first_list, second_list):
    return [x + y for x, y in zip(first_list, second_list)]
 
 
k = int(input().split()[1])
nums = [int(n) for n in input().split()]#[int(n) for n in "43251"]#list(range(1, 102))[::-1]#
###
 
mapper = []
for i in range(1, k, 2):
    mapper.append(i)
 
mapper = ([k-m for m in mapper] + [0] * (k % 2) + [-(k-m) for m in mapper][::-1])
###
 
num_moves = []
for n in range(len(nums)):
    num_moves.append((nums[n]-1 - n))
 
start_time = time.time()
###
past_moves = 0
fin_moves = []
 
last_nums = []
best = 0, 0
while True:
 
    #start_time = time.time()
    best_index = num_moves.index(max(num_moves, key=lambda i: abs(i))), 0
 
    for i in range(min(best[0]-k, 0), len(num_moves)-k+1):
 
        s = num_moves[i:i+k//2] + num_moves[i+k//2+k%2:i+k]
        d = max(s[:k//2]) # + max([abs(n) for n in s])*2
        if(d >= best[1]): #and (min(s[0:k//2]) > 0 and max(s[k//2:]) <= 0)):
            best = i, d
 
 
    
    num_moves = num_moves[:best[0]] + sum_lists(num_moves[best[0]:best[0]+k][::-1], mapper) + num_moves[best[0]+k:]
 
    past_moves += 1
    fin_moves.append(best[0]+1)
 
    id = hash(str(num_moves))
 
    if(id in last_nums):
        print("NO")
        exit()
 
    last_nums.append(id)
 
    if(num_moves == [0]*len(num_moves)):
        break

    best = 0, 0

 
print("YES")
print(len(fin_moves))
print(" ".join(map(str, fin_moves)))

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
5 2
1 2 3 4 5

correct output
YES
0

user output
YES
2
4 4

Test 2

Group: 1

Verdict: ACCEPTED

input
5 2
2 1 3 4 5

correct output
YES
1
1

user output
YES
1
1

Test 3

Group: 1

Verdict: ACCEPTED

input
20 2
6 20 18 2 16 13 19 17 8 14 11 ...

correct output
YES
366
2 3 4 5 6 7 8 9 10 11 12 13 14...

user output
YES
132
2 3 4 2 5 3 6 4 7 6 5 3 8 7 6 ...

Test 4

Group: 1

Verdict: ACCEPTED

input
100 2
100 92 62 88 12 7 43 31 19 72 ...

correct output
YES
2876
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
2438
1 2 3 4 5 6 7 8 9 1 10 2 11 3 ...

Test 5

Group: 1

Verdict: ACCEPTED

input
100 2
100 99 98 97 96 95 94 93 92 91...

correct output
YES
5248
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
4950
1 2 1 3 2 1 4 3 2 1 5 4 3 2 1 ...

Test 6

Group: 2

Verdict: ACCEPTED

input
5 3
1 2 3 4 5

correct output
YES
0

user output
YES
2
3 3

Test 7

Group: 2

Verdict: ACCEPTED

input
5 3
3 5 4 1 2

correct output
NO

user output
NO

Test 8

Group: 2

Verdict: ACCEPTED

input
5 3
5 2 1 4 3

correct output
YES
8
1 2 1 3 1 2 3 1

user output
YES
2
1 3

Test 9

Group: 2

Verdict: ACCEPTED

input
20 3
19 14 1 18 3 4 11 20 13 6 17 8...

correct output
YES
52
8 10 12 14 16 18 1 3 5 7 9 11 ...

user output
YES
42
1 3 5 4 8 7 6 2 10 9 8 4 12 11...

Test 10

Group: 2

Verdict: ACCEPTED

input
20 3
9 6 13 18 5 10 3 2 7 20 1 4 19...

correct output
YES
50
10 12 14 16 18 13 15 17 4 6 8 ...

user output
YES
34
4 6 10 8 3 12 10 5 1 14 13 12 ...

Test 11

Group: 2

Verdict:

input
500 3
53 52 21 76 25 142 5 4 83 176 ...

correct output
YES
15194
334 336 338 340 342 344 346 34...

user output
(empty)

Test 12

Group: 2

Verdict:

input
500 3
51 44 147 172 1 28 27 82 233 1...

correct output
YES
15565
366 368 370 372 374 376 378 38...

user output
(empty)

Test 13

Group: 2

Verdict:

input
500 3
75 46 179 62 221 14 67 154 89 ...

correct output
YES
15920
454 456 458 460 462 464 466 46...

user output
(empty)

Test 14

Group: 2

Verdict:

input
500 3
161 54 285 12 71 142 111 94 97...

correct output
YES
15931
408 410 412 414 416 418 420 42...

user output
(empty)

Test 15

Group: 2

Verdict: ACCEPTED

input
500 3
122 260 455 113 315 276 433 43...

correct output
NO

user output
NO

Test 16

Group: 2

Verdict:

input
500 3
499 500 497 498 495 496 493 49...

correct output
YES
62264
2 4 6 8 10 12 14 16 18 20 22 2...

user output
(empty)

Test 17

Group: 3

Verdict: ACCEPTED

input
5 4
1 2 3 4 5

correct output
YES
0

user output
YES
2
2 2

Test 18

Group: 3

Verdict:

input
5 4
5 1 2 3 4

correct output
YES
4
1 2 1 2

user output
NO

Test 19

Group: 3

Verdict:

input
500 4
58 14 107 124 4 113 24 290 56 ...

correct output
YES
15698
389 392 395 398 401 404 407 41...

user output
NO

Test 20

Group: 3

Verdict:

input
500 4
113 187 278 242 23 67 48 298 3...

correct output
YES
15004
480 481 480 482 485 488 491 49...

user output
NO

Test 21

Group: 3

Verdict:

input
500 4
5 233 199 35 213 354 11 134 30...

correct output
YES
16770
458 461 464 467 470 473 476 47...

user output
NO

Test 22

Group: 3

Verdict:

input
500 4
169 47 21 137 57 138 360 147 4...

correct output
YES
15889
497 371 372 371 373 376 379 38...

user output
NO

Test 23

Group: 3

Verdict:

input
500 4
493 409 291 313 156 443 496 40...

correct output
YES
22886
480 481 480 482 485 488 491 49...

user output
NO

Test 24

Group: 3

Verdict: ACCEPTED

input
500 4
137 99 100 226 326 298 140 340...

correct output
NO

user output
NO

Test 25

Group: 3

Verdict:

input
500 4
500 499 498 497 496 495 494 49...

correct output
YES
41458
1 2 1 2 5 8 11 14 17 20 23 26 ...

user output
NO

Test 26

Group: 4

Verdict: ACCEPTED

input
5 5
1 2 3 4 5

correct output
YES
0

user output
YES
2
1 1

Test 27

Group: 4

Verdict: ACCEPTED

input
5 5
5 4 3 2 1

correct output
YES
1
1

user output
YES
1
1

Test 28

Group: 4

Verdict:

input
500 5
145 26 285 154 147 314 141 40 ...

correct output
YES
13786
216 220 224 228 232 236 240 24...

user output
NO

Test 29

Group: 4

Verdict:

input
500 5
137 22 399 292 249 6 51 224 42...

correct output
YES
13465
456 460 464 468 472 476 480 48...

user output
NO

Test 30

Group: 4

Verdict:

input
500 5
153 52 85 100 329 60 433 468 4...

correct output
YES
13642
377 378 377 380 384 388 392 39...

user output
NO

Test 31

Group: 4

Verdict:

input
500 5
267 326 95 108 189 32 291 366 ...

correct output
YES
14639
213 214 213 216 220 224 228 23...

user output
NO

Test 32

Group: 4

Verdict: ACCEPTED

input
500 5
15 450 272 80 321 101 247 438 ...

correct output
NO

user output
NO

Test 33

Group: 4

Verdict:

input
499 5
497 498 499 496 495 494 493 49...

correct output
YES
30886
3 7 11 15 19 23 27 31 35 39 43...

user output
(empty)

Test 34

Group: 4

Verdict:

input
500 5
499 500 497 498 495 496 493 49...

correct output
YES
30919
1 4 8 12 16 20 24 28 32 36 40 ...

user output
NO