Codeforces Round #723 (Div. 2)C1+C2. Potions(dp/貪心)
阿新 • • 發佈:2021-06-10
題目大意:n瓶藥水,從左到右開始喝,喝完一瓶健康值+a[i],(a[i]有可能是負數),要保證每喝完一瓶藥水後健康值不能為負數。
問最多喝幾瓶。
題解:
解法一:DP O(n^2)
dp[i][j]表示前i瓶藥水喝j瓶的最大健康值,則轉移方程為dp[i][j]=max(dp[i-1][j-1]+a[i],dp[i-1][j])。
若dp[i][j]為負數,給它賦值一個負無窮,這樣後續狀態不會從這個狀態轉移過去。
解法二:貪心
能喝就喝,喝了變成負數了就把之前喝的最小的一瓶吐出來
#include<iostream> #include<cstring> #include<cstdio> #include<algorithm> using namespace std; #define LL long long const LL INF=1e18; int n; LL dp[2020][2020]; int a[2020]; int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); } for(int i=1;i<=n;i++) { for(int j=1;j<=i;j++) { if(i==j) dp[i][j]=dp[i-1][j-1]+a[i]; else dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]+a[i]); if(dp[i][j]<0) dp[i][j]=-INF; } } for(int j=n;j>=0;j--) { if(dp[n][j]>=0) { cout<<j;return 0; } } return 0; }
#include<iostream> #include<queue> using namespace std; int n; priority_queue<int,vector<int>,greater<int>> q; int main() { scanf("%d",&n); long long res=0; for(int i=1;i<=n;i++) { int x; scanf("%d",&x); res+=x; q.push(x); while(res<0) { int tmp=q.top(); res-=tmp; q.pop(); } } printf("%d\n",q.size()); return 0; }