rt，我写的是 pos[K][N]，$N$ 开 $4\times 10^4$ 都能过。