题目链接:
https://codeforces.com/problemset/problem/1470/A
选的标签是dp…
思路:
将k数组从大到小排序即可,用一个h记录 1-ki 个数前面最小的数的位置,如果没有被用过就拿,用过了直接加 Cki。
AC代码:
//A. Strange Birthday Party
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+10;
ll k[N],c[N],n,m;
ll d[N];
ll t;
bool st[N];
int cmp(int a,int b)
{
return a>b;
}
int main()
{
scanf("%lld",&t);
while(t--)
{
ll l=1;
memset(st,false,sizeof(st));
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%lld",&k[i]);
}
for(int i=1;i<=m;i++)
{
scanf("%lld",&c[i]);
}
sort(k+1,k+1+n,cmp);
ll ans=0;
for(int i=1;i<=n;i++)
{
int h=k[i];
if(!st[l]&&l<=h)
{
// printf("c[%d]===%lld \n",i,c[l]);
ans+=c[l];
st[l]=true;
l++;
}
else
{
// printf("c[%d]===%lld \n",i,c[h]);
ans+=c[h];
}
// cout<<ans<<endl;
}
cout<<ans<<endl;
}
return 0;
}