leetcode——【裁剪数字后查询第k小的数字】

最新推荐文章于 2024-07-05 09:21:30 发布

不瘦20斤不换头像

最新推荐文章于 2024-07-05 09:21:30 发布

阅读量179

点赞数

分类专栏： leetcode 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/super_carry/article/details/125985715

版权

leetcode 专栏收录该内容

29 篇文章 1 订阅

订阅专栏

题目

在这里插入图片描述

思路

基数排序

实现

class Solution {
public:
    vector<int> smallestTrimmedNumbers(vector<string>& nums, vector<vector<int>>& queries) {
        int len = nums.size();
        int str_len = nums[0].size();
        vector<vector<int>> vec(str_len + 1);
        for (int i = 0; i < len; ++i)
            vec[0].emplace_back(i);
        for (int i = 1; i <= str_len; ++i) {
            vector<vector<int>> B(10);
            for (int index : vec[i - 1])
                B[nums[index][str_len - i] - '0'].emplace_back(index);
            for (int j = 0; j < 10; ++j) {
                for (int in : B[j])
                    vec[i].emplace_back(in);
            }
        }
        vector<int> ret;
        for (vector<int> query : queries)
            ret.emplace_back(vec[query[1]][query[0] - 1]);
        return ret;
    }
};