LeetCode406☞根据身高重建队列
关联LeetCode题号406
本题特点
- 贪心算法
- 关于出现两个维度一起考虑的情况，（本题是身高和排行两个维度）其技巧都是确定一边然后贪心另一边，两边一起考虑，就会顾此失彼。类似LeetCode135☞分糖果
局部最优：优先按身高高的people的k来插入。插入操作过后的people满足队列属性。
全局最优：最后都做完插入操作，整个队列满足题目队列属性。
- 一开始先按照排行维度考虑排序，发现身高和排行都不能确定
- 再考虑按照身高维度排序，如果身高相同，就按照排行从小到大排序，这样就会得到一个身高按照从大小的排列顺序，
- 题目要求，每个元素前面都是身高大，所以只要根据排行的位置进行插入（头插法）即可，因为后面身高都是小于前面的，向前插入永远不会出现，该元素前有比该元素还小的的元素
python写法
class Solution:
def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
# 先按照h维度的身高顺序从高到低排序。确定第一个维度
# lambda返回的是一个元组：当-x[0](维度h）相同时，再根据x[1]（维度k）从小到大排序
people.sort(key=lambda x: (-x[0], x[1]))
que = []
# 根据每个元素的第二个维度k，贪心算法，进行插入
# people已经排序过了：同一高度时k值小的排前面。
for p in people:
que.insert(p[1], p)
return quepeople.sort(key=lambda x: (-x[0], x[1]))
#按照第一个元素的从大到小排序，如果第一个元素相同，则按照第二个元素从小到大排序
Java写法
package leetcode;
import org.junit.jupiter.api.Test;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
/**
* File Description: QueueReconstructionByHeight_406
* Author: Your Name
* Date: 2024/12/25
*/
public class QueueReconstructionByHeight_406 {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people, (a, b)->{
if (a[0] == b[0]) return a[1] - b[1];
return b[0] - a[0];
});
ArrayList<int[]> que = new ArrayList<>();
for (int[] p: people){
que.add(p[1], p);
}
int[][] q = new int[people.length][];
return que.toArray(q);
}
@Test
public void TestReconstructQueue(){
int[][] q = {{7,0}, {4,4},{7,1},{5,0},{6,1},{5,2}};
int[][] res = reconstructQueue(q);
System.out.println(Arrays.deepToString(res));
}
}
1. lambda表达式☞类的静态方法
sort(Comparator) , (a,b)->a-b 表示从小到大排序，b-a表示从大到小
Arrays.sort(people, (a, b)->{
if (a[0] == b[0]) return a[1] - b[1];
return b[0] - a[0];
})2. ArrayList转化为Arrays
ArrayList的对象方法toArray(arr) 优先将列表元素存到数组arr，否则，将为此分配相同运行时类型的新数组。
匿名类方法写成
que.toArray(new int[people.length][])3. 将Arrays转化成String输出具体元素值
一维数组使用Arrays.toString(arr)，若二维数组使用该方法，输出子层元素值
二维数组使用Arrays.deepToString(arr)