题目大意:
t个样例,每个样例输入两行,第一行输入n和m,分别表示数组的长度和可以从数组中删除多少个数字,第二行输入n个数,表示数组中的每个元素。
有下面三种操作
1.选择一个数加一
2.选择一个数减一
3.选择一个数除以2
每种操作都会消耗一点能量值。(删除m个元素不消耗能量)
问每个样例中的数组最少消耗多少点能量能使数组中的数都相同。
t<=10,m<n<=500,数组中的数字数量级为1e9。时间限制为6000ms。
解析:
铜牌题
值得注意的是,这道题的样例数最多只有10,不像其他的题动不动就1e5,并且n和m也最大只有500,最主要的是时间限制为6000ms,第一次见到这么长的时间限制,说明这个问题可能本身就没有比较高效的解法,不妨可以大胆提交。
首先,如果没有除以2操作,最终数组中的数字会变成一个相同的数,这个数一定可以是原来数组中的一个数。比如原来数组中有两个数,1和4,那么变成1 1,变成2 2,变成3 3和变成4 4所消耗的能量其实是相同的。
但是如果加上了除以2的操作,那么数组中最终会变成的同一个数,可能是原来数组中的,也可能是原来数组中的元素不停除以2得到的,比如1和4,4/2=2,变成2 2只需要两个能量。
知道了数组中的数字最终可能会变成什么结果,不妨用一个set存这些可能,然后看数组中每个数字变成这个结果所消耗的能量,然后把所需消耗最多的m个数字去掉,剩下的加起来。看看变成哪个结果所需要的能量最少。
注意要开long long
代码:
#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
#define int long long//第一次写成了define ing long long
void solve()
{
set<int> s;
int a[501],b[501];
int n,m;
cin>>n>>m;
for(int i=0;i<n;i++)
cin>>a[i];
for(int i=0;i<n;i++)
{
int t = a[i];
while(t)
{
s.insert(t);
t/=2;
}
}
int ans,res = 1e18;
for(auto it:s)
{
ans = 0;
for(int i=0;i<n;i++)
{
if(a[i]<=it)
b[i]=it-a[i];
else
{
int cnt = 0;
int t = a[i];
while(t/2>=it)
{
cnt++;
t/=2;
}
b[i] = min(t-it+cnt,it-t/2+cnt+1);
}
}
sort(b,b+n);
for(int i=0;i<n-m;i++)
ans+=b[i];
res = min(ans,res);
}
cout<<res<<'\n';
}
signed main()
{
int t;
cin>>t;
while(t--)
solve();
}