不用写堆吧，开个长度k的数组，复杂度高一点  也可以import heap啊