partial_sort使うのがいいんじゃない?