-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path21939.kt
44 lines (40 loc) · 1.25 KB
/
21939.kt
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
41
42
43
44
import java.io.*
import java.util.TreeSet
fun main(){
val bw = BufferedWriter(OutputStreamWriter(System.out))
val br = BufferedReader(InputStreamReader(System.`in`))
val n = br.readLine().toInt()
val numLevel = HashMap<Int, Int>()
val levelNum = TreeSet<Pair<Int,Int>>(compareBy({it.first},{it.second}))
repeat(n){
val (num, level) = br.readLine().split(" ").map {it.toInt()}
numLevel[num] = level
levelNum.add(Pair(level,num))
}
val m = br.readLine().toInt()
repeat(m){
val cmd = br.readLine().split(" ")
when(cmd[0]) {
"add" -> {
val num = cmd[1].toInt()
val level = cmd[2].toInt()
numLevel[num] = level
levelNum.add(Pair(level, num))
}
"recommend" -> {
if(cmd[1].toInt() == 1) {
bw.write("${levelNum.last().second}\n")
}
else
bw.write("${levelNum.first().second}\n")
}
"solved" -> {
val num = cmd[1].toInt()
levelNum.remove(Pair(numLevel[num],num))
numLevel.remove(num)
}
}
}
bw.flush()
bw.close()
}