塔子哥选数字-阿里淘天2024笔试(codefun2000)

题目链接
塔子哥选数字-阿里淘天2024笔试(codefun2000)

题目内容

塔子哥有一个长为n的数组a。他定义一个数组的权值为：数组中不同的数字个数。
塔子哥希望从数组a中选出在个数子，使得这k个数字组成的数组权值最大，请你帮帮塔子哥。

输入描述

输人包含两行
第一行两个正整效n,k(1≤k≤n≤105 )，表示数组a的长度和需要选择的数字个数
第二行n个正整数ai(1≤ai ≤109)，表示数组的元素值

输出描述

输出包含一行一个正整数，表示选出k个数字组成数组的最大权值。

样例1

输入

4 3
1 1 2 2

输出

2

提示

可以选择1,1,2，构成数组，权值为2

题解1

#include<bits/stdc++.h>
using namespace std;

int n, k, ans;
unordered_map<int,bool> hmap;

int main(){
	scanf("%d%d", &n, &k);
	for(int i = 1, u; i <= n; i++){
		scanf("%d", &u);
		if(hmap.count(u) == 0) hmap[u] = 1, ans++;
	}
	printf("%d\n", min(ans, k));
	return 0;
}