class Solution {
public:
vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
map<int,vector<int>> mp;
int N = groupSizes.size();
for(int i=0;i<N;i++){
int S = groupSizes[i];
mp[S].push_back(i);
}
vector<vector<int>> ans;
for(map<int,vector<int>>::iterator p = mp.begin();p!=mp.end();p++){
int loc = 0;
while(loc<p->second.size()){
vector<int> t;
for(int i=0;i<p->first;i++){
t.push_back(p->second[loc+i]);
}
loc += p->first;
ans.push_back(t);
}
}
return ans;
}
};
No.143 - LeetCode1282
最新推荐文章于 2022-08-16 19:57:54 发布