没看答案,遍历people中的每个人,每个人都按他们的要求排好队,这样就能得到全局最优解,运用了贪心思想。
class Solution:
def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
# 按题意的ki从低到高排,ki相同的再按身高hi从高到低排
# 这样排序是为了方便后续的插入算法
people = sorted(people, key=lambda x:(x[1],-x[0]))
queue = []
for pp in people:
if pp[1] == 0:
queue.insert(0, pp)
else:
# 按题意从已排好的队列中找到符合ki个数的比自己高的人头
# 然后插入到其中最后一个人头的后面即可
count = 0
for i in range(len(queue)):
if queue[i][0] >= pp[0]:
count += 1
if count == pp[1]:
queue.insert(i+1, pp)
break
return queue