[C++知识库]哦~唔西迪西小姐~(优先队列）

牛客练习赛

有一点理解错题意了 不是所有相同标记的格子都要跳 可以跳可以不跳
比如当前的标记为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;
}