携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第14天,点击查看活动详情
每日刷题 2022.08.12
- leetcode原题链接:leetcode.cn/problems/qu…
- 难度:中等
- 方法:结构体排序
题目
- 给你一个下标从 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 ,下标为 2 。
2. 裁剪到剩 3 个数位后,nums 没有变化。第 2 小的数字是 251 ,下标为 2 。
3. 裁剪到剩 2 个数位后,nums = ["02","73","51","14"] 。第 4 小的数字是 73 ,下标为 1 。
4. 裁剪到剩 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 ,下标为 0 的 4 视为小于下标为 3 的 4 。
2. 裁剪到剩 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
解题思路
- 最开始的时候,一直认为
js中内置的sort函数,只能给数组排序,比赛后查资料才了解到可以给结构体排序。 substr()负数,从尾部开始截取- 先把数组裁剪。由于要返回第k小数字对应的下标,也就是说排序后,需要“记忆”排序前的下标。
- 可以采用这样一种方案: 直接让数组中的值携带它的原下标信息,再调用
Array.prototype.sort()方法。具体实现类似arr.map((val, i) => ({ val, i })).sort((a, b) => a.val - b.val)。 - 原先是把值先转换为
number类型,再去比较。然而发现数据可能会超出number精度,故处理时选择了逐位比较字符串这样一种笨方法。
AC代码
/**
* @param {string[]} nums
* @param {number[][]} queries
* @return {number[]}
*/
var smallestTrimmedNumbers = function(nums, queries) {
function node (val, idx) {
this.v = val;
this.id = idx;
}
let res = [];
let cur = [], len = nums.length;
for(let q of queries) {
let n = q[1], k = q[0];
cur = [];
for(let i = 0; i < len; i++) {
let last = nums[i].substr(-q[1]);
let no = new node(last, i);
cur.push(no);
}
// 结构体排序
cur.sort((a, b) => {
if(a.v == b.v) return a.id - b.id;
return (a.v < b.v) == true ? -1 : 1;
});
res.push(cur[k - 1].id);
}
return res;
};