[LeetCode]406. Queue Reconstruction by Height

来源:互联网 发布:神曲 译本知乎 编辑:程序博客网 时间:2024/06/16 17:10

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]]

 题意及分析:给出了一些人的身高和这个人前面身高大于等于他的人数,对这些人进行排队,使得所有人的情况都得到满足。这道题的思路是(贪心算法):


  (1)首先找到身高最高的人并对他们进行排序。


  (2)然后找到身高次高的人,按照他们的前面的人数把他们插入到最高的人群中。


因此这是一个排序和插入的过程,按照身高进行降序排序,然后把身高相同的人按照k进行升序排序。每次取出身高相同的一组人,按照k值把他们插入到队列中。

public class Solution {    public int[][] reconstructQueue(int[][] people) {        Arrays.sort(people, new Comparator<int[]>(){            public int compare(int[] a, int[] b){                if(a[0]==b[0]) return a[1]-b[1];                else return b[0]-a[0];            }        });                List<int[]> res = new LinkedList<>();                for(int i=0; i<people.length; i++){            int[] peo = people[i];            res.add(peo[1], peo);        }                return res.toArray(new int[people.length][]);            }}

几个不熟悉的点

1、Java的Comparator接口实现自定义排序Arrays.sort()

详见Arrays.sort()用法

        Arrays.sort(nodes, new Comparator<Node>() {              public int compare(Node node1, Node node2) {                  if (node1.min == node2.min) {                      if (node1.sum == node2.sum)                           return node1.index - node2.index;                      else                          return node2.sum - node1.sum;                  }                  else                      return node2.min - node1.min;              }          }); 

升序是前者减去后者,降序是后者减去前者。


2、linkedList的add函数

void add(int index, Object element)

Inserts the specified element at the specified position index in this list. Throws IndexOutOfBoundsException if the specified index is out of range (index < 0 || index > size()).


3、二维数组的性质。


原创粉丝点击