和这题类似的:easy版本数据是<=2000
打了一发01背包的变式n平方碾过去了
hard的数据能支持nlogn,平方肯定过不去了
一般dp和贪心喜欢在一起
考虑贪心
对于当前的x,先尝试吃下去,同时丢到小根堆里,记住吃了多少,ans累加,cnt累加
然后发现目前的ans小于0时,反悔,把吃下去的东西里最小那个吐出来,肯定能保证ans>0了
正确性的证明:
额其实不怎么需要证明,整个过程就是一个贪心,能吃则吃,发现出大问题了吐掉不得不丢的
(一个带反悔的可爱贪心
好久没打小根堆了,甚至现场百度了一下写法..
priority_queue<int, vector<int>, greater<int> >q;
#include <iostream> #include <math.h> #include <string.h> #include <vector> #include <map> #include <queue> #include <stdio.h> #include <algorithm> #include <cstdio> using namespace std; priority_queue<int, vector<int>, greater<int> >q; int main() { // freopen("lys.in", "r", stdin); int n; int cnt=0; cin >> n; long long ans = 0; for (int i = 1; i <= n; i++) { int a; cin >> a; ans += a; cnt++; q.push(a); if (ans < 0) { int x = q.top(); q.pop(); cnt--; ans -= x; } } cout << cnt << endl; }