牛客练习赛97 哦~唔西迪西小姐~ (优先队列 贪心）

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

唔西迪西现在正处在一个冰火迷宫中，迷宫由 nnn 个格子组成，每个格子要么是冰之格，要么是火之格，唔西迪西刚开始可以选择从迷宫中任意一个开始走，走到第 iii 个位置时会得到值为 aia_iai​ 的积分。（注意：唔西迪西也可以选择一个格子都不走）

如果唔西迪西当前在冰之格，那么她可以选择一个编号大于当前格子的冰之格，跳到那里。如果唔西迪西当前在火之格，那么她可以选择一个编号大于当前格子的火之格，跳到那里。如果唔西迪西目前没有格子可以走，那么结束。同时，即使存在可以跳过去的格子，唔西迪西也可以选择在任意时刻结束。

唔西迪西想最大化她的得分，于是她学会了一个超能力，她能在比赛开始的时候改变最多 mmm 个格子的状态，即将一个格子从冰之格变成火之格或从火之格变成冰之格，改变第 iii 个格子的状态会让唔西迪西的得分减少 pip_ipi​ 。（唔西迪西改变格子的状态后才开始挑选起点开始行动，也就是说，得分分成两部分，一部分是改变格子状态的得分，一部分是走格子的得分）

你能告诉唔西迪西她最多得几分吗

输入描述:

第一行两个正整数 n,mn,mn,m (1≤n,m≤105)\ (1\leq n,m\leq 10^5) (1≤n,m≤105)。

第二行 nnn 个整数表示 aia_iai​。

第三行 nnn 个整数表示 pip_ipi​ (−105≤ai,pi≤105)\ (-10^5\leq a_i,p_i \leq 10^5) (−105≤ai​,pi​≤105)。

第四行 nnn 个整数 bib_ibi​ 为 000 或 111，000 表示这个格子是冰之格，111 表示这个格子是火之格。

输出描述:

一个数表示答案。

示例1

输入

3 2
1 15 9
2 5 7
1 0 1

输出

20

示例2

输入

10 3
80 86 57 69 59 52 94 74 9 63
8 32 77 64 53 49 22 68 27 63
1 0 0 0 1 0 0 1 1 1

输出

442

题解：

        唉，贪心都不是很会了。

好吧，其实是题目没看懂。原来只是从前往后跳看看选不选，翻不翻，考虑一下四种情况就行了。我还以为冰火格可以来回跳呢。

上代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define ll long long
#define fast std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
inline int read()
{
    int x=0,k=1; char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')k=-1;c=getchar();}
    while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*k;
}
const int maxn=2e6+10;
const int mod=998244353;
int a[maxn],b[maxn],op[maxn],n,m;
int c(int x){
	priority_queue<int>q;int sum=0;
	for(int i=1;i<=n;i++){
		if(op[i]==x){
			sum+=max(0ll,a[i]);//选不选；
			q.push(-b[i]-max(0ll,a[i]));//翻不翻；
		}
		else q.push(max(a[i]-b[i],-b[i]));//另一种翻不翻；
	}
	int k=m;
	while(!q.empty()){
		if(!k) break;		
		int f=q.top();
		q.pop();
    		if(f<=0) break;
		sum+=f;
		k--;
	}
	return sum;
}
signed main(){
	n=read();m=read();
	 for(int i=1;i<=n;i++) a[i]=read();
	 for(int i=1;i<=n;i++) b[i]=read();
	 for(int i=1;i<=n;i++) op[i]=read();
	 cout<<max(c(0),c(1));//冰火分开考虑；
}