LeetCode：6390. 滑动子数组的美丽值

news/2024/4/24 10:59:24/文章来源:https://blog.csdn.net/weixin_42204569/article/details/130350769

🍎道阻且长，行则将至。🍓

🌻算法，不如说它是一种思考方式🍀

算法专栏： 👉🏻123

一、🌱6390. 滑动子数组的美丽值

题目描述：给你一个长度为 n 的整数数组 nums ，请你求出每个长度为 k 的子数组的美丽值。
一个子数组的美丽值定义为：如果子数组中第 x 小整数是负数，那么美丽值为第 x 小的数，否则美丽值为 0 。
请你返回一个包含 n - k + 1 个整数的数组，依次表示数组中从第一个下标开始，每个长度为 k 的子数组的美丽值。
子数组指的是数组中一段连续非空的元素序列。
来源：力扣（LeetCode）
难度：中等
提示：
n == nums.length
1 <= n <= 10⁵
1 <= k <= n
1 <= x <= k
-50 <= nums[i] <= 50
示例 1：
输入：nums = [1,-1,-3,-2,3], k = 3, x = 2
输出：[-1,-2,-2]
解释：总共有 3 个 k = 3 的子数组。
第一个子数组是 [1, -1, -3] ，第二小的数是负数 -1 。
第二个子数组是 [-1, -3, -2] ，第二小的数是负数 -2 。
第三个子数组是 [-3, -2, 3] ，第二小的数是负数 -2 。
示例 2：
输入：nums = [-1,-2,-3,-4,-5], k = 2, x = 2
输出：[-1,-2,-3,-4]
解释：总共有 4 个 k = 2 的子数组。
[-1, -2] 中第二小的数是负数 -1 。
[-2, -3] 中第二小的数是负数 -2 。
[-3, -4] 中第二小的数是负数 -3 。
[-4, -5] 中第二小的数是负数 -4 。
示例 3：
输入：nums = [-3,1,2,-3,0,-3], k = 2, x = 1
输出：[-3,0,-3,-3,-3]
解释：总共有 5 个 k = 2 的子数组。
[-3, 1] 中最小的数是负数 -3 。
[1, 2] 中最小的数不是负数，所以美丽值为 0 。
[2, -3] 中最小的数是负数 -3 。
[-3, 0] 中最小的数是负数 -3 。
[0, -3] 中最小的数是负数 -3 。

🌴解题

1.暴力排序(超时)

就是遍历的时刻，把这 k 个数的字数组排序一下，选取第 x 小作为结果保存。

class Solution {public int[] getSubarrayBeauty(int[] nums, int k, int x) {int length = nums.length;int[] ans=new int[length-k+1];for (int i = 0; i < length-k+1; i++) {int num=findx(nums,i,k,x);if(num<0)ans[i]=num;else ans[i]=0;}return ans;}private static int findx(int[] nums, int i, int k, int x) {int[] tems=new int[k];for (int j = 0; j < k; j++) {tems[j]=nums[i];i++;}Arrays.sort(tems);return tems[x-1];}
}

肯定是超过时间限制了！
在这里插入图片描述

2.自排序的 Map

我们知道在 Map 集合是一种键值对的结构，我们可以使用 key 来存储数组元素，使用 value 来存储元素在子串出现的次数。其中的 TreeMap 是一种自动排序的数据结构，所以可以有这种做法：

class Solution {public int[] getSubarrayBeauty(int[] nums, int k, int x) {int[] result = new int[nums.length - k + 1];//存放返回结果Map<Integer,Integer> nmap=new TreeMap<>();// key 数字 ： value 个数for (int i = 0; i < nums.length; i++) { //遍历每一个数字if(!nmap.containsKey(nums[i])){//集合中不包含这个数字nmap.put(nums[i], 1);}else{nmap.put(nums[i], nmap.get(nums[i])+1);//集合中包含这个数字,对其 value + 1}if(i >= k-1){int j=0;for (int key: nmap.keySet()) {j+=nmap.get(key);if(j >= x) {result[i - k + 1] = key < 0 ? key : 0;break;}}if(nmap.get(nums[i - k + 1])==1){//集合中包含 1 个这个数字nmap.remove(nums[i - k + 1]);}else{nmap.put(nums[i - k + 1], nmap.get(nums[i - k + 1]) - 1);//集合中包含多个这个数字,对其 value - 1}}}return result;}
}

在这里插入图片描述

3.数组 hash

题目说明了 -50 <= nums[i] <= 50，因此只需要创建大小为 101 的数组 count 即可，数组 count 的索引表示原数组的值，数组 count 的表示出现的次数。

class Solution {public int[] getSubarrayBeauty(int[] nums, int k, int x) {int[] result = new int[nums.length - k + 1];//存放返回结果Map<Integer,Integer> nmap=new TreeMap<>();// key 数字 ： value 个数for (int i = 0; i < nums.length; i++) { //遍历每一个数字if(!nmap.containsKey(nums[i])){//集合中不包含这个数字nmap.put(nums[i], 1);}else{nmap.put(nums[i], nmap.get(nums[i])+1);//集合中包含这个数字,对其 value + 1}if(i >= k-1){int j=0;for (int key: nmap.keySet()) {j+=nmap.get(key);if(j >= x) {result[i - k + 1] = key < 0 ? key : 0;break;}}if(nmap.get(nums[i - k + 1])==1){//集合中包含 1 个这个数字nmap.remove(nums[i - k + 1]);}else{nmap.put(nums[i - k + 1], nmap.get(nums[i - k + 1]) - 1);//集合中包含多个这个数字,对其 value - 1}}}return result;}
}