程式語言 - LeetCode - C++ - 406. Queue Reconstruction by Height



題目:


解答:

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), [](auto &a, auto &b) {
            if (a[0] == b[0]) {
                return a[1] < b[1];
            }
            return a[0] > b[0];
        });

        vector<vector<int>> ans;
        for (auto& a : people) {
            ans.insert(ans.begin() + a[1], a);
        }

        return ans;
    }
};