感觉第一个没法再优化了，我用的priorityqueue，重写compare只能一直套if-else