裁剪数字后查询第 K 小的数字

·  阅读 435

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第9天,点击查看活动详情

说在前面

🎈不知道大家对于算法的学习是一个怎样的心态呢?为了面试还是因为兴趣?不管是处于什么原因,算法学习需要持续保持,今天让我们一起来看看这一道题目————裁剪数字后查询第 K 小的数字,使用哈希表可以快速得出答案。

题目描述

给你一个下标从 0 开始的字符串数组 nums ,其中每个字符串 长度相等 且只包含数字。

再给你一个下标从 0 开始的二维整数数组 queries ,其中 queries[i] = [ki, trimi] 。对于每个 queries[i] ,你需要:

将 nums 中每个数字 裁剪 到剩下 最右边 trimi 个数位。 在裁剪过后的数字中,找到 nums 中第 ki 小数字对应的 下标 。如果两个裁剪后数字一样大,那么下标 更小 的数字视为更小的数字。 将 nums 中每个数字恢复到原本字符串。 请你返回一个长度与 queries 相等的数组 answer,其中 answer[i]是第 i 次查询的结果。

提示:

  • 裁剪到剩下 x 个数位的意思是不断删除最左边的数位,直到剩下 x 个数位。
  • nums 中的字符串可能会有前导 0 。  

示例 1:

输入:nums = ["102","473","251","814"], queries = [[1,1],[2,3],[4,2],[1,2]]
输出:[2,2,1,0]
解释:
1. 裁剪到只剩 1 个数位后,nums = ["2","3","1","4"] 。最小的数字是 1 ,下标为 22. 裁剪到剩 3 个数位后,nums 没有变化。第 2 小的数字是 251 ,下标为 23. 裁剪到剩 2 个数位后,nums = ["02","73","51","14"] 。第 4 小的数字是 73 ,下标为 14. 裁剪到剩 2 个数位后,最小数字是 2 ,下标为 0 。
   注意,裁剪后数字 "02" 值为 2复制代码

示例 2:

输入:nums = ["24","37","96","04"], queries = [[2,1],[2,2]]
输出:[3,0]
解释:
1. 裁剪到剩 1 个数位,nums = ["4","7","6","4"] 。第 2 小的数字是 4 ,下标为 3 。
   有两个 4 ,下标为 04 视为小于下标为 342. 裁剪到剩 2 个数位,nums 不变。第二小的数字是 24 ,下标为 0复制代码

提示:

  • 1 <= nums.length <= 100
  • 1 <= nums[i].length <= 100
  • nums[i] 只包含数字。
  • 所有 nums[i].length 的长度 相同 。
  • 1 <= queries.length <= 100
  • queries[i].length == 2
  • 1 <= ki <= nums.length
  • 1 <= trimi <= nums[0].length

思路分析

首先我们要理解题目的意思,题目参数中的queries为一个二维数组,其中queries[i]为一次询问queries[i][1]表示将数字裁剪到剩下queries[i][1]位,queries[i][0]表示将nums中的数字裁剪成剩下queries[i][1]位后获取第queries[i][0]小数字的下标。

  • 如示例一
nums = ["102","473","251","814"]
queries[2][4,2]
首先我们将nums裁剪剩下queries[2][1]位,也即是2位,得到nums = ["02","73","51","14"]
其中第queries[2][0]小,也即是第4小的数字为73,其下标为1
所以我们应该要返回的结果为1
复制代码

理解了题目之后我们便可以开始解题了,这里我使用一个哈希表将每一种情况都记录下来,后面询问的时候就可以直接从哈希表中取出结果,具体如下:

  • 获取截取后的数字列表

key为截取数字长度,value[0]为数字的下标,value[1]为截取的数字。

let map = {};
for(let i = 0; i < nums.length; i++){
 let str = '';
 for(let j = nums[i].length - 1; j >= 0; j--){
     let ind = nums[i].length - j;
     str = nums[i][j] + str;
     if(!map[ind]) map[ind] = [];
     map[ind].push([i,str]);
 }
}
复制代码

得到结果如下

{ 
'1': [ [ 0, '2' ], [ 1, '3' ], [ 2, '1' ], [ 3, '4' ] ], 
'2': [ [ 0, '02' ], [ 1, '73' ], [ 2, '51' ], [ 3, '14' ] ], 
'3': [ [ 0, '102' ], [ 1, '473' ], [ 2, '251' ], [ 3, '814' ] ] 
} 
复制代码
  • 将数字列表进行排序 将的得到的结果按规则进行排序,数字从小到大进行排序,如果两个裁剪后数字一样大,那么下标 更小 的数字视为更小的数字。
for(let key in map){
     map[key] = map[key].sort((a,b)=>{
        if(a[1] == b[1]) return a[0] - b[0];
        return a[1].localeCompare(b[1]);
     })
 }
复制代码

得到结果如下

{ 
'1': [ [ 2, '1' ], [ 0, '2' ], [ 1, '3' ], [ 3, '4' ] ], 
'2': [ [ 0, '02' ], [ 3, '14' ], [ 2, '51' ], [ 1, '73' ] ], 
'3': [ [ 0, '102' ], [ 2, '251' ], [ 1, '473' ], [ 3, '814' ] ] 
}
复制代码
  • 遍历取出答案
let res = [];
for(let i = 0; i < queries.length; i++){
    let value = map[queries[i][1]][queries[i][0] - 1][0];
    res.push(value);
}
复制代码

完整AC代码如下:

AC代码

/**
 * @param {string[]} nums
 * @param {number[][]} queries
 * @return {number[]}
 */
 var smallestTrimmedNumbers = function(nums, queries) {
     let map = {};
     for(let i = 0; i < nums.length; i++){
         let str = '';
         for(let j = nums[i].length - 1; j >= 0; j--){
             let ind = nums[i].length - j;
             str = nums[i][j] + str;
             if(!map[ind]) map[ind] = [];
             map[ind].push([i,str]);
         }
     }
     for(let key in map){
         map[key] = map[key].sort((a,b)=>{
            if(a[1] == b[1]) return a[0] - b[0];
            return a[1].localeCompare(b[1]);
         })
     }
     let res = [];
     for(let i = 0; i < queries.length; i++){
         let value = map[queries[i][1]][queries[i][0] - 1][0];
         res.push(value);
     }
     return res;
};
复制代码

说在后面

🎉这里是JYeontu,喜欢算法,GDCPC打过卡;热爱羽毛球,大运会打过酱油。毕业一年,两年前端开发经验,目前担任H5前端开发,算法业余爱好者,有空会刷刷算法题,平时喜欢打打羽毛球🏸 ,也喜欢写些东西,既为自己记录📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解🙇,写错的地方望指出,定会认真改进😊,在此谢谢大家的支持,我们下文再见🙌。

分类:
前端
分类:
前端
收藏成功!
已添加到「」, 点击更改