Time: 26 ms (27.61%), Space: 13.4 MB (42.88%) - LeetHub

Author: sachin10101998

1282-group-the-people-given-the-group-size-they-belong-to/1282-group-the-people-given-the-group-size-they-belong-to.cpp

@@ -0,0 +1,15 @@
+class Solution {
+public:
+    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
+        vector<vector<int>> res;
+        vector<vector<int>> gz (groupSizes.size()+1);
+        for(int i=0;i<groupSizes.size();i++){
+            gz[groupSizes[i]].push_back(i);
+            if(gz[groupSizes[i]].size()==groupSizes[i]){
+                res.push_back({});
+                swap(res.back(),gz[groupSizes[i]]);
+            }
+        }
+        return res;
+    }
+};