题目意思:
Petya有n个朋友,每个朋友都要送一个礼物,他给每个朋友定好了k值(给朋友前k个礼物中的一个),以及买好了m个礼物(礼物金额从小到大给你排好了),每个礼物的价格都定好了。Petya给每个朋友的礼物有两种选择:1.给他前k个礼物中的一个,但是每个礼物只能送一次。2.给他第k个礼物的美金。问你如何将送出去的礼物花的钱最小。
解题思路:
1.将朋友的k值进行从大到小排序,因为k值大的可选范围大,但第k个礼物的金额也就更贵,我们可以将k值小的给他第k个礼物的金额来达到送出的礼物总金额最小。
2.礼物从最小金额开始送,送出去的礼物标记为不能送出的状态,没送出去的可以送给k值较大的朋友。k值较小的朋友直接拿美金。
注意:
如何让k值小的朋友拿美金呢,分两种情况:
1.如果第k个礼物被人拿走,意味着前k个礼物都被拿走了,并且礼物金额从小到大排序,他就只能拿第k个礼物的美金。
2.如果拿走礼物的钱比第k个礼物更贵,那就拿第k个礼物的美金。
#include<iostream>
#include<algorithm>
using namespace std;
long long int p[300001];
struct node{
long long int num,value;//num标记礼物是否被拿走,value为价值
}g[300001];//定义数组结构
bool cmp(int x,int y)//排序
{
return x>y;
}
int main()
{
long long int t,n,m,i,sum,ming,ss;
cin>>t;
while(t--)
{
sum=0;ming=1000000000;ss=1;
cin>>n>>m;
for(i=1;i<=n;i++)
{
cin>>p[i];
}
for(i=1;i<=m;i++)
{
cin>>g[i].value;
g[i].num=1;//礼物标记为未拿走
}
sort(p+1,p+n+1,cmp);//对朋友的k值排序
for(i=1;i<=n;i++)
{
if(g[i].num!=-1)//如果礼物没被拿走
{
if(g[ss].value<g[p[i]].value)//如果拿走礼物更划算
{
sum+=g[ss].value;g[ss].num=-1;ss++;//就将礼物标记为拿走状态
}
else//如果拿走礼物不划算
{
sum+=g[p[i]].value;//那就拿钱
}
}
else//如果礼物被拿走
{
sum+=g[p[i]].value;//那就拿钱
}
}
cout<<sum<<endl;
}
return 0;
}