leetcode题目:406. 根据身高重建队列(贪心算法)

1. 题目分析

leetcode题目:406. 根据身高重建队列(贪心算法)_第1张图片
本题有两个维度,h和k,看到这种题目一定要想如何确定一个维度,然后在按照另一个维度重新排列。
我们不能h和k一起考虑,这样会顾此失彼,我们必须先确定一个维度,h或者k。
如果按照k来从小到大排序,排完之后,会发现k的排列并不符合条件,身高也不符合条件,两个维度哪一个都没确定下来。

  • 思路:
    按照k从高到底排序,这样前面的k一定高于后面的,在重新创建一个队列,遍历排序后的数组,再重新插入就行了。

  • 我们拿第一个样例来说
    排序完的people:
    [[7,0], [7,1], [6,1], [5,0], [5,2],[4,4]]
    插入的过程:
    插入[7,0]:[[7,0]]
    插入[7,1]:[[7,0],[7,1]]
    插入[6,1]:[[7,0],[6,1],[7,1]]
    插入[5,0]:[[5,0],[7,0],[6,1],[7,1]]
    插入[5,2]:[[5,0],[7,0],[5,2],[6,1],[7,1]]
    插入[4,4]:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]

那么在以后遇见类似的题型,是不是就能考虑到先确定其中一个维度的思想呢?

2. 代码实现

2.1. Python代码

class Solution(object):
    def reconstructQueue(self, people):
        """
        :type people: List[List[int]]
        :rtype: List[List[int]]
        """
        people = sorted(people,reverse=True,key=lambda x: (x[0], -x[1]))
        res = []
        for i in range(len(people)):
            if people[i][1] < i:
                res.insert(people[i][1],people[i])
            else:
                res.append(people[i])
        return res

2.2. Java代码

class Solution {
     
    public int[][] reconstructQueue(int[][] people) {
     
        int[][] res = new int[people.length][2];
        people = sort(people);
        res[0] = people[0];
        for(int i = 1;i < people.length;i++){
     
            if(i > people[i][1]){
     
                for(int j = i;j > people[i][1];j--){
     
                    res[j] = res[j - 1];
                }
                res[people[i][1]] = people[i];
            }else{
     
                res[i] = people[i];
            }
        }
        return res;
    }

    public int[][] sort(int[][] people){
     
        for(int i = 0;i < people.length;i++){
     
            int max = i;
            for(int j = i + 1;j < people.length;j++){
     
                if(people[j][0] > people[max][0]){
     
                    max = j;
                }
                if(people[j][0] == people[max][0] && people[j][1] < people[max][1]){
     
                    max = j;
                }
            }
            if(max != i){
     
                int[] t = people[max];
                people[max] = people[i];
                people[i] = t;
            }
        }
        return people;
    }
}

你可能感兴趣的:(leetcode,算法,leetcode,贪心算法)