Skip to content
本页目录

0406-根据身高重建队列

题目描述

https://leetcode.cn/problems/queue-reconstruction-by-height/

假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ,前面 正好 有 ki 个身高大于或等于 hi 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。

示例 1:

输入:people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释:
编号为 0 的人身高为 5 ,没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ,没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ,有 2 个身高更高或者相同的人排在他前面,即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
编号为 4 的人身高为 4 ,有 4 个身高更高或者相同的人排在他前面,即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。

示例 2:

输入:people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
输出:[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]

提示:

  • 1 <= people.length <= 2000
  • 0 <= hi <= 10^6
  • 0 <= ki < people.length
  • 题目数据确保队列可以被重建

思路【官方】

https://leetcode.cn/problems/queue-reconstruction-by-height/solution/gen-ju-shen-gao-zhong-jian-dui-lie-by-leetcode-sol/

csharp
public class Solution {
    public int[][] ReconstructQueue(int[][] people) {
    	Array.Sort(people, (a,b)=>{
    		if(a[0] != b[0]){
    			return a[0].CompareTo(b[0]);	
    		}
    		else{
    			return b[1].CompareTo(a[1]);
    		}
    		
    	});
    	int n = people.Length;
    	int[][] ans = new int[n][];

    	for(int i=0; i < people.Length; i++){
    		int spaces = people[i][1] + 1;
    		for(int j=0; j<n;j++){
    			if(ans[j] == null){
    				spaces--;
    				if(spaces == 0){
    					ans[j] = new int[]{people[i][0],people[i][1]};
    					break;
    				}
	    		}
    		}    		
    	}
    	return ans;
    }
}

思路2:降升排序+插入

people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]

先按照身高反序排列,然后按照第二个升序排列得到:

[[7,0],[7,1],[6,1],[5,0],[5,2],[4,4]]

此时我们按照第二个坐标往新的结果集去插入对应的位置

第1次,结果集为空, 插入 [7,0] 第2次,我们插入到1的位置,这样保证前面有1个是在我面前的 第3次,我们还是插入到1的位置,此时结果集中前面一个数移到了位置2,为什么这次插入不会影响前面的数据呢?因为我们按照降序排列,之前的 k 只是统计了前面大于等于自己身高的,小于的不统计,所以我们直接插入到目标位置,不会影响最终结果 依次插入后面的数据,得到最终结果。 注意:第二列需要按照升序排列,否则,比如先插入了[5,2] 在插入 [5,0]的时候,那么 [5,2]将被推到[5,3]的位置了。

csharp
public class Solution {
    public int[][] ReconstructQueue(int[][] people) {
        //套路:这种两个元素的排列,可以试试第一个升序第二个反序,或者第一个反序第二个升序排列的方式
        //本题中,先按照第一个降序,第二个升序排列,因为第二个是表示前面升高大于等于自己的,所以这样排序可以减少插入次数
        Array.Sort(people,(a,b)=>{
            if(a[0] != b[0]){
                return b[0].CompareTo(a[0]); //身高反序
            }
            else{
                return a[1].CompareTo(b[1]); //前面身高个数降序
            }
        });

        List<int> list = new List<int>(); //用list先表示顺序,插入数据
        for(int i=0; i<people.Length; i++){
            list.Insert(people[i][1],i);
        }

        int[][] result = new int[people.Length][]; //输出最后结果
        for(int i=0; i<people.Length; i++){
            int[] p = people[list[i]];
            result[i] = p;
        }
        return result;
    }
}

复习:20220620

csharp
public class Solution {
    public int[][] ReconstructQueue(int[][] people) {
        //题目中 ki 表示前面有 ki 个身高大于或 小于 hi 的个数
        //换句话说,身高比自己矮的插在自己前面试无所谓的
        //将数组按照身高顺序排列,然后再按照次数插入到index出,这样不影响最终的结果
        //注意相等的时候,要按照升序排列,否则插入的时候,会打乱次序
        Array.Sort(people, (a,b)=>{
            if(a[0] != b[0]){
                return b[0].CompareTo(a[0]);
            }
            else{
                return a[0].CompareTo(b[0]);
            }
        });
        //先在身高都排列好了,将结果依次按照索引顺序插入数组中去,此处用 List 实现
        List<int> list = new List<int>();
        for(int i=0; i<people.Length; i++){
            list.Insert(people[i][1],i); //将索引插入数组,比如 [7,0],[6,1],[5,0] 插入的时候,第一个插入0处,第二个插入1处,第三个插入0出,此时 第一个往后移动一位,但是因为5比他们小,所以不影响后面的个数值。 
        }
        //输出结果
        int[][] result = new int[people.Length][];
        for(int i=0; i<list.Count; i++){
            int index = list[i];
            result[i] = new int[]{people[index][0],people[index][1]};
        }
        return result;
    }
}

Released under the MIT License.