406. Queue Reconstruction by Height
最后更新
二刷
15-Jan-2016
排座问题，记得以前是H难度的，现在改成M了。。
总结一句话，先按身高排，高的在前面；身高一样的情况下，K小的在前面。
BB半天说白了一句话，H大的在顶端，H一样的K小的在顶端。
根据这句话设立PQ的判断规则就行了。。
这样排的原因是因为出QUEUE的时候，后出的因为H小，不会影响前面的人。 而H一样的时候，虽然会影响前面同身高的，但是肯定K小的放前面，因为他能容忍比自己>=h的人更少。
最后的问题就是返还是list of int[]，所以得找个办法记录PERSON在原先队列的位置= =麻烦了不少。
中间数组和LIST来回换很容易把自己弄晕= =||，有了这个做法也不想看别的做法了，拿到一个OFFER之后，我已经是一条咸鱼了。
说好的拿到这个OFFER决不负自己呢？
Time: O(NlogN)
Space: O(N)
public class Solution {
public class Person {
int height;
int k;
int index;
public Person(int h, int k, int i) {
this.height = h;
this.k = k;
this.index = i;
}
}
public int[][] reconstructQueue(int[][] people) {
PriorityQueue<Person> pq = new PriorityQueue<>(new Comparator<Person>() {
public int compare(Person a, Person b) {
if (a.height != b.height) {
return Integer.compare(b.height, a.height);
} else {
return Integer.compare(a.k, b.k);
}
}
});
for (int i = 0; i < people.length; i++) {
pq.offer(new Person(people[i][0], people[i][1], i));
}
// contain index of each person in the input array.
List<Integer> refList = new ArrayList<>();
while (!pq.isEmpty()) {
Person tempPerson = pq.poll();
refList.add(tempPerson.k, tempPerson.index);
}
int[][] res = new int[people.length][2];
for (int i = 0; i < refList.size(); i++) {
int pos = refList.get(i);
res[i][0] = people[pos][0];
res[i][1] = people[pos][1];
}
return res;
}
}
一刷
25-Sep-2016
一开始backtrack，设计了很多剪枝情况，还是TLE了
。。后来用PQ做的。
其实上面DFS做到一半的时候意识到应该用PQ做，但是不确定会不会TLE，就继续了，然后果然TLE了。。
PQ的做法和剪枝的判断标准是一样的，只不过PQ不是遍历，比兼职纯遍历要快。
想象一下排座位，h是身高，k是某个小朋友最多允许几个人挡住他（貌似不是很恰当）。
对于单个小朋友来说，只有比他高才会挡住他，所以我们先按K的标准排最高的一批小朋友，这样一来后面批次的小朋友不会影响自己，因为永远不可能挡住自己。
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
[7,0]和[7,1]先排好。
同一批次里，K小的在前面，这个很好理解。。
第二批小朋友是身高为6的，[6,1]，他要保证只有1个人挡住他，他坐到刚才2个人之间就行了。
[7,0] [6,1] [7,1]
第三批是身高为5的，[5,0]和[5,2]，[5,0]想近距离观察美女老师，而目前在坐的所有人都比他高，他就跑去第一排了。至于[5,2]，性欲不那么旺盛，被2个人挡住也无所谓，于是坐到2个人后面。
[5,0] [7,0] [6,1] [7,1]
[5,0] [7,0] [5,2] [6,1] [7,1]
然后是最后一个小朋友[4,4]，他可能人比较耿直，正人君子，上课的时候心无旁骛，不需要看美女老师，所以前面4个人无所谓，他跑到4个人后面
[5,0] [7,0] [5,2] [6,1] [4,4] [7,1]
BB半天说白了一句话，H大的在顶端，H一样的K小的在顶端。
根据这句话设立PQ的判断规则就行了。。
这样排的原因是因为出QUEUE的时候，后出的因为H小，不会影响前面的人。 而H一样的时候，虽然会影响前面同身高的，但是肯定K小的放前面，因为他能容忍比自己>=h的人更少。
public class Solution {
public class People
{
public int h;
public int k;
public int i; //index of this people in people[][]
public People(int h, int k, int i)
{
this.h = h;
this.k = k;
this.i = i;
}
}
public class compare implements Comparator<People>
{
public int compare(People p1, People p2)
{
if(p1.h!=p2.h) return p2.h-p1.h;
else return p1.k-p2.k;
}
}
public int[][] reconstructQueue(int[][] people)
{
if(people.length == 0) return new int[0][0];
PriorityQueue<People> pq1 = new PriorityQueue(new compare());
for(int i = 0; i < people.length;i++)
{
pq1.add(new People(people[i][0],people[i][1],i));
}
//index of people in people[][]
List<Integer> l = new ArrayList<Integer>();
while(!pq1.isEmpty())
{
People p = pq1.poll();
l.add(p.k,p.i);
}
int[][] res = new int[people.length][2];
for(int i = 0; i < l.size();i++)
{
res[i][0] = people[l.get(i)][0];
res[i][1] = people[l.get(i)][1];
}
return res;
}
}
