406. Queue Reconstruction by Height

最新推荐文章于 2023-11-18 01:25:37 发布

一头食量超大的小猪

最新推荐文章于 2023-11-18 01:25:37 发布

阅读量268

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/sinat_17451213/article/details/53443714

版权

leetcode 专栏收录该内容

118 篇文章 0 订阅

订阅专栏

Suppose you have a random list of people standing in a queue. Each person is described by a pair of integers (h, k), where h is the height of the person and k is the number of people in front of this person who have a height greater than or equal to h. Write an algorithm to reconstruct the queue.

Note:
The number of people is less than 1,100.

Example

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

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

分析：首先按照身高从小到大的顺序排序，身高相同的则根据其前面大于等于其的人数从小到大的顺序排序；

身高相同的其相对顺序已确定；

首先将身高最高的索引列表，依次将身高第二高/第三高...的根据其前面比其高的人数插入列表，插入索引就是其前面比其高的人数。

实现时，先排序，同时存储在列表中的是索引；然后再根据此列表对应的数组中的第二个元素值作为索引插入新列表中，此时新列表中的索引即是排好序的索引。

public class Solution {
    public int[][] reconstructQueue(int[][] people) {
        int n=people.length;
        List<Integer> list=new ArrayList<Integer>();//保存索引
        list.add(0);
        //sort the array，先按照身高从大到小， 然后再按其前面的人数从小到大
        for(int i=1;i<n;i++){
            int j=0;
           for(;j<i;j++){
               int index=list.get(j);
               if(people[i][0]>people[index][0]){
                    list.add(j,i);
                    break;
               }else if(people[i][0]==people[index][0]){
                   int k=j;
                   while(k<list.size()&&people[i][0]==people[list.get(k)][0]){
                       if(people[i][1]<people[list.get(k)][1]){
                            list.add(k,i);
                            break;
                       }
                       k++;
                   }
                    if(k==list.size())
                        list.add(i);
                    else if(people[i][0]!=people[list.get(k)][0]){
                        list.add(k,i);
                    }
                    break;
                  
               }
           }
           if(j==list.size())
                list.add(i);
        }
        
        List<Integer> list1=new ArrayList<Integer>();
        
        // 根据排序后的元素的位置插入
        int i=0;
        for(;i<n;i++){
            int index=people[list.get(i)][1];
            if(index<list1.size())
                list1.add(index,list.get(i));
            else
                list1.add(list.get(i));
        }
        
        int[][] res=new int[n][2];
        for(i=0;i<n;i++){
            res[i][0]=people[list1.get(i)][0];
            res[i][1]=people[list1.get(i)][1];
            
        }
        return res;
    }
}