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;
}
}