CSES - Aalto Competitive Programming 2024 - wk6 - Mon - Results
Submission details
Task:Terrible security
Sender:louaha1
Submission time:2024-10-07 17:01:34 +0300
Language:Python3 (CPython3)
Status:READY
Result:
Test results
testverdicttime
#1ACCEPTED0.02 sdetails
#2ACCEPTED0.02 sdetails
#3ACCEPTED0.02 sdetails
#4ACCEPTED0.02 sdetails
#5ACCEPTED0.02 sdetails
#6ACCEPTED0.02 sdetails
#7ACCEPTED0.02 sdetails
#8ACCEPTED0.02 sdetails
#9ACCEPTED0.02 sdetails
#10ACCEPTED0.02 sdetails
#11ACCEPTED0.02 sdetails
#12ACCEPTED0.02 sdetails
#13ACCEPTED0.02 sdetails
#14ACCEPTED0.02 sdetails
#15ACCEPTED0.02 sdetails
#16ACCEPTED0.02 sdetails
#17ACCEPTED0.02 sdetails
#18ACCEPTED0.02 sdetails
#19ACCEPTED0.02 sdetails
#20ACCEPTED0.02 sdetails
#21ACCEPTED0.02 sdetails
#22ACCEPTED0.02 sdetails
#23ACCEPTED0.02 sdetails
#24ACCEPTED0.02 sdetails
#25ACCEPTED0.02 sdetails
#26ACCEPTED0.02 sdetails
#27ACCEPTED0.02 sdetails
#28ACCEPTED0.02 sdetails
#29ACCEPTED0.02 sdetails
#30ACCEPTED0.02 sdetails
#31ACCEPTED0.02 sdetails
#32ACCEPTED0.02 sdetails
#33ACCEPTED0.02 sdetails
#34ACCEPTED0.02 sdetails
#35ACCEPTED0.02 sdetails
#36ACCEPTED0.02 sdetails
#37ACCEPTED0.02 sdetails
#38ACCEPTED0.02 sdetails
#39ACCEPTED0.02 sdetails
#40ACCEPTED0.02 sdetails
#41ACCEPTED0.02 sdetails
#42ACCEPTED0.02 sdetails
#43ACCEPTED0.02 sdetails
#44ACCEPTED0.02 sdetails
#45ACCEPTED0.02 sdetails
#46ACCEPTED0.02 sdetails
#47ACCEPTED0.02 sdetails
#48ACCEPTED0.02 sdetails
#49ACCEPTED0.02 sdetails
#50ACCEPTED0.02 sdetails
#51ACCEPTED0.02 sdetails
#52ACCEPTED0.02 sdetails
#53ACCEPTED0.02 sdetails
#54ACCEPTED0.02 sdetails
#55ACCEPTED0.02 sdetails
#56ACCEPTED0.02 sdetails
#57ACCEPTED0.02 sdetails
#58ACCEPTED0.02 sdetails
#59ACCEPTED0.12 sdetails
#60ACCEPTED0.03 sdetails
#61ACCEPTED0.10 sdetails
#62ACCEPTED0.02 sdetails
#63ACCEPTED0.13 sdetails
#64ACCEPTED0.03 sdetails
#65ACCEPTED0.11 sdetails
#66ACCEPTED0.31 sdetails
#67ACCEPTED0.34 sdetails
#68ACCEPTED0.36 sdetails
#69--details
#70--details
#71--details
#72--details
#73--details
#74--details

Code

def prison_escape(n, q, keys, queries):
    locker_map = {i + 1: keys[i] for i in range(n)}
    
    results = []
    
    for x in queries:
        visited = set() 
        stack = [x] 
        while stack:
            current = stack.pop()
            if current not in visited:
                visited.add(current)
                stack.append(locker_map[current])  
        results.append(len(visited)) 
    
    return results

n, q = map(int, input().split())
keys = list(map(int, input().split()))
queries = list(map(int, input().split()))

answers = prison_escape(n, q, keys, queries)

print(" ".join(map(str, answers)))

Test details

Test 1

Verdict: ACCEPTED

input
1 1


correct output

user output
1

Test 2

Verdict: ACCEPTED

input
2 2
2 1 
1 1 

correct output
2 2 

user output
2 2

Test 3

Verdict: ACCEPTED

input
2 1
1 2 

correct output

user output
1

Test 4

Verdict: ACCEPTED

input
3 1
2 1 3 

correct output

user output
1

Test 5

Verdict: ACCEPTED

input
3 1
2 1 3 

correct output

user output
2

Test 6

Verdict: ACCEPTED

input
3 3
1 2 3 
2 3 3 

correct output
1 1 1 

user output
1 1 1

Test 7

Verdict: ACCEPTED

input
3 1
1 2 3 

correct output

user output
1

Test 8

Verdict: ACCEPTED

input
3 2
1 3 2 
3 1 

correct output
2 1 

user output
2 1

Test 9

Verdict: ACCEPTED

input
4 3
1 2 4 3 
1 1 3 

correct output
1 1 2 

user output
1 1 2

Test 10

Verdict: ACCEPTED

input
4 3
2 3 1 4 
1 2 2 

correct output
3 3 3 

user output
3 3 3

Test 11

Verdict: ACCEPTED

input
4 3
1 2 3 4 
4 1 3 

correct output
1 1 1 

user output
1 1 1

Test 12

Verdict: ACCEPTED

input
4 1
3 4 1 2 

correct output

user output
2

Test 13

Verdict: ACCEPTED

input
4 4
2 1 3 4 
3 4 1 1 

correct output
1 1 2 2 

user output
1 1 2 2

Test 14

Verdict: ACCEPTED

input
4 3
2 1 3 4 
2 1 1 

correct output
2 2 2 

user output
2 2 2

Test 15

Verdict: ACCEPTED

input
5 5
1 3 2 4 5 
3 5 3 4 4 

correct output
2 1 2 1 1 

user output
2 1 2 1 1

Test 16

Verdict: ACCEPTED

input
5 1
1 3 2 5 4 

correct output

user output
1

Test 17

Verdict: ACCEPTED

input
5 5
1 2 5 4 3 
3 3 3 2 2 

correct output
2 2 2 1 1 

user output
2 2 2 1 1

Test 18

Verdict: ACCEPTED

input
5 1
1 2 3 4 5 

correct output

user output
1

Test 19

Verdict: ACCEPTED

input
5 5
3 4 5 1 2 
3 1 5 5 4 

correct output
5 5 5 5 5 

user output
5 5 5 5 5

Test 20

Verdict: ACCEPTED

input
5 2
1 4 3 5 2 
5 5 

correct output
3 3 

user output
3 3

Test 21

Verdict: ACCEPTED

input
5 2
1 5 3 2 4 
5 1 

correct output
3 1 

user output
3 1

Test 22

Verdict: ACCEPTED

input
5 4
1 2 3 4 5 
3 5 2 3 

correct output
1 1 1 1 

user output
1 1 1 1

Test 23

Verdict: ACCEPTED

input
5 2
3 1 2 4 5 
5 2 

correct output
1 3 

user output
1 3

Test 24

Verdict: ACCEPTED

input
5 1
1 2 4 3 5 

correct output

user output
1

Test 25

Verdict: ACCEPTED

input
10 5
6 2 4 5 3 1 7 8 10 9 
7 7 4 5 3 

correct output
1 1 3 3 3 

user output
1 1 3 3 3

Test 26

Verdict: ACCEPTED

input
10 10
7 2 9 4 1 5 6 10 3 8 
2 3 1 4 2 4 4 7 4 10 

correct output
1 2 4 1 1 1 1 4 1 2 

user output
1 2 4 1 1 1 1 4 1 2

Test 27

Verdict: ACCEPTED

input
10 5
2 1 3 8 7 10 5 4 6 9 
4 4 2 3 7 

correct output
2 2 2 1 2 

user output
2 2 2 1 2

Test 28

Verdict: ACCEPTED

input
10 9
1 7 5 4 3 2 8 6 9 10 
5 9 1 2 1 3 3 1 1 

correct output
2 1 1 4 1 2 2 1 1 

user output
2 1 1 4 1 2 2 1 1

Test 29

Verdict: ACCEPTED

input
10 10
3 4 5 6 7 8 9 10 1 2 
6 2 10 9 8 7 7 6 3 2 

correct output
5 5 5 5 5 5 5 5 5 5 

user output
5 5 5 5 5 5 5 5 5 5

Test 30

Verdict: ACCEPTED

input
10 5
1 4 6 5 9 7 3 8 2 10 
1 7 4 8 4 

correct output
1 3 4 1 4 

user output
1 3 4 1 4

Test 31

Verdict: ACCEPTED

input
10 1
9 5 1 4 2 10 3 7 6 8 

correct output

user output
1

Test 32

Verdict: ACCEPTED

input
10 10
1 3 2 5 4 6 7 10 8 9 
4 6 3 6 1 1 5 3 1 5 

correct output
2 1 2 1 1 1 2 2 1 2 

user output
2 1 2 1 1 1 2 2 1 2

Test 33

Verdict: ACCEPTED

input
10 6
8 6 3 10 1 5 7 2 4 9 
9 3 5 1 7 5 

correct output
3 1 5 5 1 5 

user output
3 1 5 5 1 5

Test 34

Verdict: ACCEPTED

input
10 1
1 4 2 3 7 5 6 8 9 10 

correct output

user output
3

Test 35

Verdict: ACCEPTED

input
100 78
8 2 4 12 9 1 7 6 3 11 10 5 15 ...

correct output
11 8 3 2 1 1 1 5 8 5 1 5 6 1 4...

user output
11 8 3 2 1 1 1 5 8 5 1 5 6 1 4...
Truncated

Test 36

Verdict: ACCEPTED

input
100 6
7 17 9 15 1 5 6 19 16 8 20 12 ...

correct output
3 7 3 5 10 7 

user output
3 7 3 5 10 7

Test 37

Verdict: ACCEPTED

input
100 35
2 1 3 4 5 8 15 14 6 7 11 9 12 ...

correct output
5 12 2 2 12 13 13 12 7 12 3 12...

user output
5 12 2 2 12 13 13 12 7 12 3 12...

Test 38

Verdict: ACCEPTED

input
100 98
1 2 18 5 7 15 11 19 6 3 17 10 ...

correct output
16 7 14 11 4 11 14 14 14 16 14...

user output
16 7 14 11 4 11 14 14 14 16 14...
Truncated

Test 39

Verdict: ACCEPTED

input
100 91
3 4 5 6 7 8 9 10 11 12 13 14 1...

correct output
50 50 50 50 50 50 50 50 50 50 ...

user output
50 50 50 50 50 50 50 50 50 50 ...
Truncated

Test 40

Verdict: ACCEPTED

input
100 57
6 5 3 2 4 1 14 17 36 41 32 22 ...

correct output
1 2 12 2 1 7 5 10 3 10 10 29 1...

user output
1 2 12 2 1 7 5 10 3 10 10 29 1...
Truncated

Test 41

Verdict: ACCEPTED

input
100 58
11 92 21 40 79 26 20 7 53 16 3...

correct output
18 68 68 18 68 18 18 68 68 68 ...

user output
18 68 68 18 68 18 18 68 68 68 ...
Truncated

Test 42

Verdict: ACCEPTED

input
100 47
23 5 11 3 16 21 13 10 8 18 15 ...

correct output
15 15 10 10 10 10 1 5 10 10 7 ...

user output
15 15 10 10 10 10 1 5 10 10 7 ...
Truncated

Test 43

Verdict: ACCEPTED

input
100 54
42 57 3 73 56 79 95 17 60 23 8...

correct output
72 8 72 18 72 72 18 72 1 72 72...

user output
72 8 72 18 72 72 18 72 1 72 72...
Truncated

Test 44

Verdict: ACCEPTED

input
100 8
16 14 30 28 23 1 26 25 8 22 13...

correct output
21 3 1 1 6 21 1 4 

user output
21 3 1 1 6 21 1 4

Test 45

Verdict: ACCEPTED

input
200 32
8 2 4 12 9 1 7 6 3 11 10 5 15 ...

correct output
11 1 15 15 11 5 5 8 9 11 15 3 ...

user output
11 1 15 15 11 5 5 8 9 11 15 3 ...

Test 46

Verdict: ACCEPTED

input
200 30
7 17 9 15 1 5 6 19 16 8 20 12 ...

correct output
13 1 5 2 10 4 11 4 13 11 6 4 1...

user output
13 1 5 2 10 4 11 4 13 11 6 4 1...

Test 47

Verdict: ACCEPTED

input
200 87
2 1 3 4 5 8 15 14 6 7 11 9 12 ...

correct output
6 6 3 14 3 12 3 2 4 2 2 9 11 1...

user output
6 6 3 14 3 12 3 2 4 2 2 9 11 1...
Truncated

Test 48

Verdict: ACCEPTED

input
200 75
1 2 18 5 7 15 11 19 6 3 17 10 ...

correct output
10 8 5 7 9 9 10 7 10 7 2 1 3 1...

user output
10 8 5 7 9 9 10 7 10 7 2 1 3 1...
Truncated

Test 49

Verdict: ACCEPTED

input
200 181
3 4 5 6 7 8 9 10 11 12 13 14 1...

correct output
100 100 100 100 100 100 100 10...

user output
100 100 100 100 100 100 100 10...
Truncated

Test 50

Verdict: ACCEPTED

input
200 63
6 5 10 2 8 1 3 4 9 12 11 7 82 ...

correct output
19 22 19 39 1 19 7 39 6 4 27 3...

user output
19 22 19 39 1 19 7 39 6 4 27 3...
Truncated

Test 51

Verdict: ACCEPTED

input
200 99
11 92 21 164 79 144 20 7 105 1...

correct output
138 38 138 138 138 138 138 2 1...

user output
138 38 138 138 138 138 138 2 1...
Truncated

Test 52

Verdict: ACCEPTED

input
200 52
31 25 18 4 38 19 46 29 8 44 21...

correct output
57 57 13 42 24 57 15 42 57 42 ...

user output
57 57 13 42 24 57 15 42 57 42 ...
Truncated

Test 53

Verdict: ACCEPTED

input
200 95
42 57 3 119 56 173 95 191 60 1...

correct output
144 144 144 32 144 144 144 144...

user output
144 144 144 32 144 144 144 144...
Truncated

Test 54

Verdict: ACCEPTED

input
200 11
16 67 30 28 23 1 38 61 64 22 1...

correct output
63 16 63 47 47 12 47 3 6 47 1 

user output
63 16 63 47 47 12 47 3 6 47 1

Test 55

Verdict: ACCEPTED

input
1000 411
8 2 4 12 9 1 7 6 3 11 10 5 15 ...

correct output
3 11 1 8 9 1 1 6 12 1 1 7 4 14...

user output
3 11 1 8 9 1 1 6 12 1 1 7 4 14...
Truncated

Test 56

Verdict: ACCEPTED

input
1000 186
7 17 9 15 1 5 6 19 16 8 20 12 ...

correct output
13 3 2 10 17 4 12 3 1 7 9 8 11...

user output
13 3 2 10 17 4 12 3 1 7 9 8 11...
Truncated

Test 57

Verdict: ACCEPTED

input
1000 952
2 1 3 4 5 8 15 14 6 7 11 9 12 ...

correct output
5 19 7 7 5 4 1 14 11 8 3 6 12 ...

user output
5 19 7 7 5 4 1 14 11 8 3 6 12 ...
Truncated

Test 58

Verdict: ACCEPTED

input
1000 975
1 2 18 5 7 15 11 19 6 3 17 10 ...

correct output
1 5 3 4 5 6 9 8 1 15 10 15 4 2...

user output
1 5 3 4 5 6 9 8 1 15 10 15 4 2...
Truncated

Test 59

Verdict: ACCEPTED

input
1000 901
3 4 5 6 7 8 9 10 11 12 13 14 1...

correct output
500 500 500 500 500 500 500 50...

user output
500 500 500 500 500 500 500 50...
Truncated

Test 60

Verdict: ACCEPTED

input
1000 294
6 22 37 28 14 56 55 49 29 12 4...

correct output
30 231 231 290 290 290 244 2 1...

user output
30 231 231 290 290 290 244 2 1...
Truncated

Test 61

Verdict: ACCEPTED

input
1000 635
446 970 21 164 995 689 783 426...

correct output
681 681 681 681 83 83 681 198 ...

user output
681 681 681 681 83 83 681 198 ...
Truncated

Test 62

Verdict: ACCEPTED

input
1000 59
217 160 224 49 38 19 46 194 8 ...

correct output
27 52 333 333 23 333 333 27 33...

user output
27 52 333 333 23 333 333 27 33...
Truncated

Test 63

Verdict: ACCEPTED

input
1000 981
451 340 330 827 925 173 494 19...

correct output
634 634 149 634 634 634 634 63...

user output
634 634 149 634 634 634 634 63...
Truncated

Test 64

Verdict: ACCEPTED

input
1000 250
16 272 312 28 23 258 144 194 2...

correct output
186 70 88 70 186 246 70 88 246...

user output
186 70 88 70 186 246 70 88 246...
Truncated

Test 65

Verdict: ACCEPTED

input
100000 15183
8 2 4 12 9 1 7 6 3 11 10 5 15 ...

correct output
3 2 8 9 6 4 18 7 13 8 3 6 10 1...

user output
3 2 8 9 6 4 18 7 13 8 3 6 10 1...
Truncated

Test 66

Verdict: ACCEPTED

input
100000 80738
7 17 9 15 1 5 6 19 16 8 20 12 ...

correct output
9 2 2 2 8 8 7 4 12 9 5 4 14 6 ...

user output
9 2 2 2 8 8 7 4 12 9 5 4 14 6 ...
Truncated

Test 67

Verdict: ACCEPTED

input
100000 88540
2 1 3 4 5 8 15 14 6 7 11 9 12 ...

correct output
4 13 2 2 3 1 15 11 4 2 12 13 1...

user output
4 13 2 2 3 1 15 11 4 2 12 13 1...
Truncated

Test 68

Verdict: ACCEPTED

input
100000 95145
1 2 18 5 7 15 11 19 6 3 17 10 ...

correct output
12 8 5 2 1 9 13 17 5 14 1 11 5...

user output
12 8 5 2 1 9 13 17 5 14 1 11 5...
Truncated

Test 69

Verdict:

input
100000 90064
3 4 5 6 7 8 9 10 11 12 13 14 1...

correct output
50000 50000 50000 50000 50000 ...

user output
(empty)

Test 70

Verdict:

input
100000 98544
1197 327 641 3158 2038 1048 56...

correct output
51677 37867 37867 216 51677 51...

user output
(empty)

Test 71

Verdict:

input
100000 57490
70439 82301 64482 96766 51484 ...

correct output
11976 62820 62820 62820 2808 6...

user output
(empty)

Test 72

Verdict:

input
100000 72506
16265 2606 21836 8643 4486 321...

correct output
44533 16055 11222 11222 16055 ...

user output
(empty)

Test 73

Verdict:

input
100000 64645
47136 80135 73346 78143 9204 6...

correct output
13351 40626 40626 40626 40414 ...

user output
(empty)

Test 74

Verdict:

input
100000 6200
17945 16321 396 7538 19397 151...

correct output
115 13754 19010 13754 9095 137...

user output
(empty)