1. 程式人生 > >Leetcode:406. 根據身高重建佇列

Leetcode:406. 根據身高重建佇列

假設有打亂順序的一群人站成一個佇列。 每個人由一個整數對(h, k)表示,其中h是這個人的身高,k是排在這個人前面且身高大於或等於h的人數。 編寫一個演算法來重建這個佇列。

注意:
總人數少於1100人。

示例

輸入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

輸出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

解題思路:

特殊排序。先對人群進行排序,排序方式如下:

static bool cmp(pair<int, int> a, pair<int, int>b) {
        if (a.first == b.first) return a.second < b.second;
        return a.first > b.first;
    }

  1. 接著,獲取第一個元素到新的陣列res中。
  2. 遍歷之後的元素poeple[pos]逐個插入res中。
  3. 要明白,排序之後後面的元素升高都小於或等於res中的所有人,poeple[pos].k的值,可以確定poeple[pos]應該位於res陣列的第k+1個,因此在對應位置插入即可。
  4. 仔細思考,上面這個排序,它巧妙地避免了res中找不到k+1位置地尷尬,簡直玄學。

C++程式碼
class Solution {
public:
    static bool cmp(pair<int, int> a, pair<int, int>b) {
        if (a.first == b.first) return a.second < b.second;
        return a.first > b.first;
    }
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        sort(people.begin(), people.end(), Solution::cmp);
        int size = people.size(),i;
        if (size <= 1) return people;
        vector<pair<int, int>>  res;
        res.push_back(people[0]);
        for (i = 2; i <= size; i++) {
            res.insert(res.begin()+people[i - 1].second, people[i - 1]);
        }
        return res;
    }
};