【leetcode】根据身高重建队列

题目要求

根据身高重建队列

• 假设有打乱顺序的一群人战成一个队列，每个人由一个整数对(h,k)表示，
• 其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。
• 编写一个算法来重建这个队列。(总人数少于1100人)

核心思路

二维数组，利用ArrayList来进行遍历。

完整代码如下

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

/**
 * 根据身高重建队列
 * 假设有打乱顺序的一群人战成一个队列，每个人由一个整数对(h,k)表示，
 * 其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。
 * 编写一个算法来重建这个队列。(总人数少于1100人)
 */
public class Solution {
	public static int[][] rebuildHeight(int[][] rebuildHeight){
		//排列
//		Arrays.sort(rebuildHeight, new Comparator<int[]>(){
//			public int compare(int[] a, int[] b) {
//				if(a[0] != b[0]) {
//					return b[0] - a[0];
//				}else {
//					return a[1]-b[1];
//				}
//			}
//		});
		Arrays.sort(rebuildHeight, (a, b) -> (a[0] == b[0] ? a[1] - b[1] : b[0] - a[0]));
		//利用了lamda表达式，功能如上。
		List<int[]> res = new ArrayList<>();
		for(int[] p : rebuildHeight) {
			res.add(p[1],p);
		}
		return res.toArray(new int[res.size()][]);
	}
	
	public static void main(String[] args) {
		int[][] height = {{7,0},{4,4},{7,1},{5,0},{6,1},{5,2}};
		rebuildHeight(height);
		for(int i = 0; i < height.length; i++) {
			System.out.println(Arrays.toString(height[i]));
		}
	}
}