题意:
就是给你一个数组,然后定义这个数组的权值是,diff(a)-mex(a)。也就是数组a中不同元素的个数-没有出现的最小值。然后你最多操作m次,每次操作可以让一个数变为任意一个数。问你数组权值最小是多少。
思考:
刚开始到感觉就像以前做过的那种,在set里操作就好。问题就是如何转化数字,首先那些数组出现多次的肯定是我转化的备用人选。然后每个数剩下一个,放在set里,然后看看是否能到大now这个mex,如果不行,那么转化,从备用人选拿,先不管拿谁。然后如果备用人选用完了。那么肯定也要转化,那么就在set里面最大值来转化过来,因为mex可能往后走,但是绝对走不到最大值。所以这个最大值也成为了被转化的人。那么最后如果转化的次数不到m,那么答案肯定是0,都没用到那么多,肯定是0了。如果到m了,那么那些所有被转化的人,肯定是一个集合里面的人,从中选择一些。选择哪些?首先如果这个人的数字<=mex,也就是对diff没有影响的不看。如果这个人>mex,那么就可能我要转化的人。把可能要转化的数值和他出现的次数放到队列里,按次数小的在上。那么转化的时候,尽量把这个数值的所有人都转化掉,这样才能让diff更小。所以贪心一下即可。
这个题和上个题一样,就是先别管要转化谁,要转化的都放起来。最后在贪心操作一般就可以了。
代码:
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;
int T,n,m,k;
int va[N];
void solve()
{
set<int > s;
vector<int > v;
for(int i=1;i<=n;i++)
{
if(s.count(va[i])) v.pb(va[i]);
else s.insert(va[i]);
}
int now = -1,cnt = 0,sum = 0;
vector<int > ned;
while(s.size())
{
int minn = *s.begin();
if(minn==now+1)
{
now += 1;
s.erase(minn);
}
else
{
if(cnt<m)
{
if(sum<v.size()) sum++,cnt++,now++;
else
{
if(s.size())
{
cnt++,now++;
int maxn = *s.rbegin();
s.erase(maxn);
ned.pb(maxn);
}
}
}
else break;
}
}
if(cnt==m)
{
map<int,int> mp;
for(auto t:s) mp[t]++;
for(auto t:v) if(t>now) mp[t]++;
for(auto t:ned) mp[t]++;
priority_queue<PII,vector<PII>,greater<PII> > q;
for(auto t:mp) q.push({t.se,t.fi});
int ans = q.size();
while(q.size())
{
if(cnt>=q.top().fi)
{
cnt -= q.top().fi;
ans--;
}
q.pop();
}
cout<<ans<<"\n";
}
else cout<<0<<"\n";
}
signed main()
{
IOS;
cin>>T;
while(T--)
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>va[i];
solve();
}
return 0;
}
总结:
多多思考。