根据身高重建队列-贪心406-python

没看答案,遍历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

你可能感兴趣的:(动态规划,贪心,leetcode,贪心算法,python)