Create 2343.Query-Kth-Smallest-Trimmed-Number.cpp

a925907195 · Jul 17, 2022 · 535a03a · 535a03a
1 parent 635499d
commit 535a03a
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/Greedy/2343.Query-Kth-Smallest-Trimmed-Number/2343.Query-Kth-Smallest-Trimmed-Number.cpp b/Greedy/2343.Query-Kth-Smallest-Trimmed-Number/2343.Query-Kth-Smallest-Trimmed-Number.cpp
@@ -0,0 +1,37 @@
+class Solution {
+public:
+    vector<int> smallestTrimmedNumbers(vector<string>& nums, vector<vector<int>>& queries) 
+    {
+        int m = nums.size(), n = nums[0].size();
+        vector<vector<int>>ans(n+1, vector<int>(m));
+
+        for (int i=0; i<m; i++)
+            ans[0][i] = i;
+
+        for (int t=1; t<=n; t++)
+        {
+            vector<vector<int>>bucket(10);
+            for (int i=0; i<m; i++)
+            {
+                int idx = ans[t-1][i];                
+                bucket[nums[idx][n-t]-'0'].push_back(idx);
+            }
+
+            int i = 0;
+            for (int b=0; b<10; b++)            
+                for (int idx: bucket[b])
+                {
+                    ans[t][i] = idx;
+                    i++;
+                }
+        }
+
+        vector<int>rets;
+        for (auto& q: queries)
+        {
+            rets.push_back(ans[q[1]][q[0]-1]);
+        }            
+
+        return rets;
+    }
+};