ลองพิจารณาว่าเรามีรายชื่อคนที่ยืนต่อคิวแบบสุ่ม ถ้าแต่ละคนอธิบายด้วยจำนวนเต็มคู่ (h, k) โดยที่ h คือความสูงและ k คือจำนวนคนที่อยู่ข้างหน้าซึ่งมีความสูงมากกว่าหรือเท่ากับ h เราต้องกำหนดวิธีหนึ่งเพื่อสร้างคิวใหม่ ดังนั้นหากอาร์เรย์ที่กำหนดเป็นเหมือน [[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]] ผลลัพธ์จะเป็น [[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -
- จัดเรียงอาร์เรย์ที่กำหนดตามกลยุทธ์การเปรียบเทียบต่อไปนี้
- ถ้า a[0] =b[0] ให้คืนค่า a[1]> b[1] มิฉะนั้น ให้คืนค่า a[0]
- ถ้า a[0] =b[0] ให้คืนค่า a[1]> b[1] มิฉะนั้น ให้คืนค่า a[0]
- สร้างเวกเตอร์ที่เรียกว่า ans
- สำหรับ i ในขนาดช่วงของอาร์เรย์ที่กำหนดลงไปที่ 0
- แทรก (องค์ประกอบแรกของ ans + p[i, 1], p[i]) ลงในอาร์เรย์ ans
- คืนสินค้า
ตัวอย่าง
ให้เราดูการใช้งานต่อไปนี้เพื่อทำความเข้าใจ −
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<vector<auto> > v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << "[";
for(int j = 0; j <v[i].size(); j++){
cout << v[i][j] << ", ";
}
cout << "],";
}
cout << "]"<<endl;
}
bool cmp(vector <int> a, vector <int> b){
if(a[0] == b[0])return a[1] > b[1];
return a[0] < b[0];
}
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& p) {
sort(p.begin(), p.end(), cmp);
vector < vector <int> > ans;
for(int i = p.size()-1; i>=0; i--){
ans.insert(ans.begin() + p[i][1], p[i]);
}
return ans;
}
};
main(){
Solution ob;
vector<vector<int>> v = {{7,0}, {4,4}, {7,1}, {5,0}, {6,1}, {5,2}};
print_vector(ob.reconstructQueue(v));
} อินพุต
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
ผลลัพธ์
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
