forked from ngthanhtrung23/CompetitiveProgramming
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathH.kt
More file actions
40 lines (34 loc) · 841 Bytes
/
H.kt
File metadata and controls
40 lines (34 loc) · 841 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import java.util.*
fun mul(a: IntArray, b: IntArray, n: Int): IntArray {
var res = IntArray(n, {i: Int -> 0})
for (i in 0..n-1) {
res[i] = a[b[i]]
}
return res
}
fun inv(a: IntArray, n: Int): IntArray {
var res = IntArray(n, {i: Int -> 0})
for (i in 0..n-1) {
res[a[i] - 1] = i
}
return res
}
fun main(args: Array<String>) {
val (sn, sk) = readLine()!!.split(' ')
val n = sn.toInt()
var k = sk.toLong()
var a = readLine()!!.split(' ').map(String::toInt).toIntArray()
a = inv(a, n)
var res = IntArray(n, { i: Int -> i+1})
while (k > 0) {
if (k % 2 == 1L) {
res = mul(res, a, n)
}
k /= 2
a = mul(a, a, n)
}
for (x in res) {
print("$x ")
}
println("")
}