查看: 311|回复: 3
收起左侧

[Leetcode] 悬赏解答 leetcode 992 为什么我的解法超时了

|只看干货
uncle Max | 显示全部楼层 |阅读模式
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   97% (2894)
 
 
2% (86)    👎
60小米
看了 lee哥在讨论区的答案,然后自己写了写。所有test case 都能过,但是不知道为什么超时了。看起来应该和lee哥的逻辑差不多
以下是我的代码

  1. class Solution {
  2.     public int subarraysWithKDistinct(int[] nums, int k) {
  3.         return helper(nums, k) - helper(nums, k - 1);
  4.     }
  5.     public int helper(int[] nums, int k){
  6.         int count = 0;
  7.         Map<Integer, Integer> map = new HashMap<>();


  8.         for(int i = 0, j = 0; i < nums.length; i ++){
  9.             map.put(nums[i], i);
  10.             while(map.size() > k){
  11.                 int leftMost = Integer.MAX_VALUE;
  12.                 for(int num : map.values()){
  13.                     leftMost = Math.min(num, leftMost);
  14.                 }
  15.                 map.remove(nums[leftMost]);
  16.                 j = leftMost + 1;
  17.             }
  18.            count += i - j + 1;
  19.         }
  20.         return count;
  21.     }
  22. }
复制代码
以下是lee哥的答案

  1. public int subarraysWithKDistinct(int[] A, int K) {
  2.         return atMostK(A, K) - atMostK(A, K - 1);
  3.     }
  4.     int atMostK(int[] A, int K) {
  5.         int i = 0, res = 0;
  6.         Map<Integer, Integer> count = new HashMap<>();
  7.         for (int j = 0; j < A.length; ++j) {
  8.             if (count.getOrDefault(A[j], 0) == 0) K--;
  9.             count.put(A[j], count.getOrDefault(A[j], 0) + 1);
  10.             while (K < 0) {
  11.                 count.put(A[i], count.get(A[i]) - 1);
  12.                 if (count.get(A[i]) == 0) K++;
  13.                 i++;
  14.             }
  15.             res += j - i + 1;
  16.         }
  17.         return res;
  18.     }
复制代码

最佳答案

查看完整内容

大致看了下,lee的做法是map保存频数,利用sliding window在不同的num个数大于k时,收左边界。这样的话,收左边界的操作最多总共耗时O(n)。 你的算法是map保存出现的数的最新位置,这样当大于k时,遍历map的values,也就是下标位置,找到最小的移除,然后记录新产生的subarray。 问题在于遍历map的values本身是O(k)操作,这样的话总的收左边界的操作耗时是O(n*k)。题设n和k都为2 * 10^4,n*k会超时。

评分

参与人数 1大米 +3 收起 理由
14417335 + 3 很有用的信息!

查看全部评分


上一篇:苹果6个月力扣高频题
下一篇:LeetCode每日一题全勤 活动终止
我已全仓 2021-12-1 06:29:23 | 显示全部楼层
本楼:   0% (0)
 
 
0% (0)  
全局: 👍   100% (27)
 
 
0% (0)    👎
大致看了下,lee的做法是map保存频数,利用sliding window在不同的num个数大于k时,收左边界。这样的话,收左边界的操作最多总共耗时O(n)。

你的算法是map保存出现的数的最新位置,这样当大于k时,遍历map的values,也就是下标位置,找到最小的移除,然后记录新产生的subarray。

问题在于遍历map的values本身是O(k)操作,这样的话总的收左边界的操作耗时是O(n*k)。题设n和k都为2 * 10^4,n*k会超时。

评分

参与人数 2大米 +5 收起 理由
14417335 + 3 给你点个赞!
uncle Max + 2 给你点个赞!

查看全部评分

回复

使用道具 举报

 楼主| uncle Max 2021-12-2 01:06:31 | 显示全部楼层
本楼:   0% (0)
 
 
0% (0)  
全局: 👍   97% (2894)
 
 
2% (86)    👎
我已全仓 发表于 2021-11-30 17:12
大致看了下,lee的做法是map保存频数,利用sliding window在不同的num个数大于k时,收左边界。这样的话,收 ...

仔细看了看 非常感谢您的讲解 醍醐灌顶

评分

参与人数 1大米 +1 收起 理由
我已全仓 + 1 很好的题 谢谢分享!

查看全部评分

回复

使用道具 举报

本楼:   0% (0)
 
 
0% (0)  
全局: 👍   100% (27)
 
 
0% (0)    👎
uncle Max 发表于 2021-12-01 09:06:31
仔细看了看 非常感谢您的讲解 醍醐灌顶
很好的题 谢谢分享!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 注册账号
隐私提醒:
  • ☑ 禁止发布广告,拉群,贴个人联系方式:找人请去🔗同学同事飞友,拉群请去🔗拉群结伴,广告请去🔗跳蚤市场,和 🔗租房广告|找室友
  • ☑ 论坛内容在发帖 30 分钟内可以编辑,过后则不能删帖。为防止被骚扰甚至人肉,不要公开留微信等联系方式,如有需求请以论坛私信方式发送。
  • ☑ 干货版块可免费使用 🔗超级匿名:面经(美国面经、中国面经、数科面经、PM面经),抖包袱(美国、中国)和录取汇报、定位选校版
  • ☑ 查阅全站 🔗各种匿名方法

本版积分规则

>
快速回复 返回顶部 返回列表