牛客练习赛
有一点理解错题意了 不是所有相同标记的格子都要跳 可以跳可以不跳
比如当前的标记为1 如果有一个标记为0的格子 a{i】<0 p【i】也i小于零 那我们可以选择反转这个格子 但是不去跳这个格子 我们就可以获得一个-p【i】
相当于一个正数 我就是这一点理解错了 然后将a【i】大于零小于零的情况都分开考虑存到优先队列里面 然后选择大于0的反转就可以了啦!!
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#include<queue>
//默认的是从大到小排序
const int N=1e5+10;
typedef long long ll;
int a[N],b[N],p[N],n,m;
ll getanser(int flag)
{
// printf("++ %d\n",flag);
priority_queue<int>q;
int i;
ll sum=0;
for(i=1; i<=n; i++)
{
if(b[i]==flag)
{
if(a[i]>0)
{
sum+=a[i];
q.push(-a[i]-p[i]);//如果当前格子反转后获得的积分比现在的积分大 我们当然也可以选择反转它
}
else
q.push(-p[i]);//小于0的我们可以不跳它 但是可以获得p的积分
}
else
{
if(a[i]>0)
q.push(a[i]-p[i]); 跟上面的同理
else
q.push(-p[i]);
}
}
// printf("*** %d\n",sum);
int cnt=0;
while(!q.empty()&&cnt<m)//选择m个大于0的值反转就可以啦
{
int w=q.top();
// printf("+++ %d\n",w);
q.pop();
if(w>0)
{
sum+=w;
// q.pop();
cnt++;
}
}
return sum; //一开始忘记返回值了 结果一直不对 这里也要注意了
}
int main()
{
int i,j,k;
cin>>n>>m;
for(i=1; i<=n; i++)
scanf("%d",&a[i]);
for(i=1; i<=n; i++)
scanf("%d",&p[i]);
for(i=1; i<=n; i++)
scanf("%d",&b[i]);
ll ans=0;
ll x=getanser(1);
ll y=getanser(0);
ans=max(x,y);
// printf("%d %d\n",x,y);
printf("%d\n",ans);
return 0;
}