前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1481. 不同整数的最少数目(计数+排序+贪心)

LeetCode 1481. 不同整数的最少数目(计数+排序+贪心)

作者头像
Michael阿明
发布2020-07-13 15:19:29
5270
发布2020-07-13 15:19:29
举报
文章被收录于专栏:Michael阿明学习之路

1. 题目

给你一个整数数组 arr 和一个整数 k 。现需要从数组中恰好移除 k 个元素,请找出移除后数组中不同整数最少数目。

代码语言:javascript
复制
示例 1:
输入:arr = [5,5,4], k = 1
输出:1
解释:移除 1 个 4 ,数组中只剩下 5 一种整数。

示例 2:
输入:arr = [4,3,1,1,3,3,2], k = 3
输出:2
解释:先移除 4、2 ,然后再移除两个 1 中的任意 1 个或者三个 3 中的任意 1 个,
最后剩下 1 和 3 两种整数。
 
提示:
1 <= arr.length <= 10^5
1 <= arr[i] <= 10^9
0 <= k <= arr.length

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/least-number-of-unique-integers-after-k-removals 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • map计数
  • 把数量排序
  • 取出前面个数少的,能尽可能多的取出不同的种类
代码语言:javascript
复制
class Solution {	//C++
public:
    int findLeastNumOfUniqueInts(vector<int>& arr, int k) {
    	unordered_map<int,int> m;
    	int i;
    	for(i = 0; i < arr.size(); ++i)
    		m[arr[i]]++;
    	vector<int> count(m.size());
    	i = 0;
    	for(auto it = m.begin(); it != m.end(); ++it)
    		count[i++] = it->second;
    	sort(count.begin(), count.end());
    	i = 0;
    	while(k > 0)
    	{
    		k -= count[i];
    		if(k >= 0)
    			i++;
    	}
    	return count.size()-i;
    }
};

432 ms 61.2 MB

代码语言:javascript
复制
class Solution:# py3
    def findLeastNumOfUniqueInts(self, arr: List[int], k: int) -> int:
        d = {}
        for i in arr:
            if i in d:
                d[i] += 1
            else:
                d[i] = 1
        count = list(d.values())
        # count.sort()
        count = sorted(count)
        i = 0
        for c in count:
            k -= c
            if k >= 0:
                i += 1
            if k <= 0:
                break
        return len(count) - i

120 ms 26.8 MB

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/06/14 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档