根据C ++所属的小组规模对人员进行分组
假设有n个人，其ID的范围在0到n-1之间，并且每个人都完全属于一个组。我们有长度为n的数组groupSizes。该数组表示每个人所属的组的大小，我们必须找到存在的组以及每个组所包含的人员ID。
假设输入为-[3,3,3,3,3,1,3]，则输出为[[5]，[0、1、2]，[3、4、6]]，其他可能解决方案可以是[[2,1,6]，[5]，[0,4,3]]或[[5]，[0,6,2]，[4,3,1]]
为了解决这个问题，我们将遵循以下步骤-
创建一张映射
对于0到给定数组a–1的大小的范围
将我插入m[g[i]]
创建一个称为res的矩阵
对于映射m中的每个元素
将i数组的第j个元素插入temp
如果temp的大小=我的关键
然后将temp插入res作为新行
清除临时数组
对于范围从0到i出现的列表大小的j
返回资源。
让我们看下面的实现以更好地理解-
示例
#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;
}
class Solution {
public:
vector<vector<int>> groupThePeople(vector<int>& g) {
map <int, vector <int> > m;
for(int i = 0; i < g.size(); i++){
m[g[i]].push_back(i);
}
vector < vector <int> > res;
map <int, vector <int> > :: iterator i = m.begin();
vector <int> temp;
while(i != m.end()){
for(int j = 0; j < i->second.size(); j++){
temp.push_back(i->second[j]);
if(temp.size() == i->first){
res.push_back(temp);
temp.clear();
}
}
i++;
}
return res;
}
};
main(){
vector<int> v = {3,3,3,3,3,1,3};
Solution ob;
print_vector(ob.groupThePeople(v));
}输入值
[3,3,3,3,3,1,3]
输出结果
[[5, ],[0, 1, 2, ],[3, 4, 6, ],]