CodeForces - 1526C2 - Potions (Hard Version) （ 优先队列 + 贪心 ）

题目链接：点击进入

题目

题意

n 个瓶子，每个瓶子都有价值（ 有正有负，正的加血，负的减血 ），初始生命值 0 ，问在保证过程中血量不小于 0 的情况下，最多可以选择多少个瓶子（ 每个瓶子可以选择也可以不选择 ）

思路

对于价值为正的瓶子，我们肯定选择，若瓶子的价值为负：
若我们喝下去生命值不会变为负数，我们选择喝下去。
若我们喝下去生命值会变为负数，那么我们将此时的这瓶水，与之前喝过的、价值为负数的 且 价值最低的瓶子做比较，
若此时的价值更低，那么我们不选择这个瓶子；
若此时价值更高，那么我们选择这瓶，将之前选择的用来做对比的那瓶水抛弃掉（不喝）
所以，我们需要用一个 优先队列 来维护之前喝过的、价值为负数的、价值最低的瓶子价值。
剩下的就是贪心统计答案了。

代码

// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
//#include<bits/stdc++.h>
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<bitset>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
#define max(a,b) a>b?a:b
#define min(a,b) a<b?a:b
#define int long long
//#define double long double
//#ifndef ONLINE_JUDGE  freopen("data.in.txt","r",stdin);
//freopen("data.out.txt","w",stdout); #endif //Îļþ¶ÁÈ¡ 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int Mod=998244353;
const double eps=1e-9;
const int N=26;
const int mod=51123987;
const int maxn=1e6+10;

/*--------------------------------------------*/ 
inline int read()
{
    int data=0,w=1; char ch=0;
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();
    return data*w;
}
/*--------------------------------------------*/

int t,n,m,a[maxn],x;
string s;
priority_queue<int,vector<int>,greater<int>>q;

signed main()																						
{	
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	int sum=0,ans=0;
	for(int i=1;i<=n;i++)
	{
		if(a[i]>=0) sum+=a[i],ans++;
		else if(sum+a[i]>=0)
		{
			ans++;
			sum+=a[i];
			q.push(a[i]);
		}
		else
		{
			if(q.size()&&q.top()<a[i])
			{
				sum-=q.top();
				q.pop();
				sum+=a[i];
				q.push(a[i]);
			}
		}
	} 
	cout<<ans<<endl;
	return 0;
}