Source Code

#include <bits/stdc++.h>

using namespace std;
int arr[100001];
map <int,int>mp;
int main()
{
	int x,k,ans=0;
	cin>>x>>k;
	for(int i=0;i<x;i++){
		cin>>arr[i];
		mp [arr[i]]++;
	}
	if(mp[0]==0){
		cout<<0;
		return 0;
	}
	
	for(int i=0;i<100001;i++){
		if(!mp[i])break;
		if(mp[0]<mp[i]){
		ans+=mp[0];
			x-=mp[0];
		}
		else {
			ans+=mp[i];
			x-=mp[i];
		}
		if(ans > x){
			break;
		}
		
	}
	if (k> mp[0]){
			ans -= k-mp[0];
		}
	cout<<ans;
	
	
	
			
    return 0;
}
Copy
Bitar The Handy Man fakher20
GNU G++17
3 ms
304 KB
Wrong Answer