1. 程式人生 > 其它 >Codeforces Round #723 (Div. 2)C1+C2. Potions(dp/貪心)

Codeforces Round #723 (Div. 2)C1+C2. Potions(dp/貪心)

傳送門

題目大意: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;
}