我认为没有必要什么“第一行一个整数 n,k······”，直接想AtCoder那样，

n k
a_1 a_2 ... a_n

这样简单明了，看着也更舒服。