用C++按身高重建佇列
設我們有一群隨機的人站在佇列裡。如果每個人都由一對整數(h、k)進行描述,其中 h 是身高,k 是他前面身高比他大或者等於他的人數,我們必須定義一種方法來重建佇列。因此,如果給定陣列為[[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] < b[0]
- 建立一個名為ans的向量
- 對於i在給定陣列大小到0的範圍內
- 將(ans + p[i, 1]的第一個元素,p[i])插入到ans陣列中
- 返回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]]
廣告
資料結構
網路
RDBMS
作業系統
Java
iOS
HTML
CSS
Android
Python
C語言程式設計
C++
C#
MongoDB
MySQL
Java
PHP