Cf1155d dp+优先队列

枚举乘x区间的开头i;dp[2][i]表示从i往左往右最大sum;对于每个i要找max (sum(i~j)*x+dp[1][j+1]),考虑优先队列维护

i到i-1时是把所有的sum加上a[i]*x,可以外部用个变量sum维护变化,每次插入一个dp[1][j]-sum;

#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;

int n;
long long x;
long long a[300005];

long long dp[2][300005];

int main() {

    while(~scanf("%d%lld",&n,&x))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]);
        }

        memset(dp,0,sizeof(dp));
        long long ans=0;
        for(int i=1;i<=n;i++)
        {
            dp[0][i]=max(0ll,a[i]+dp[0][i-1]);
            dp[1][n+1-i]=max(0ll,a[n+1-i]+dp[1][n+1-i+1]);
            ans=max(ans,max(dp[0][i],dp[1][n+1-i]));
        }

        priority_queueq;

        long long sum=0;
        for(int i=n;i>=1;i--)
        {
            q.push(dp[1][i+1]-sum);
            sum+=a[i]*x;

            ans=max(ans,dp[0][i-1]+q.top()+sum);
        }
        printf("%lld\n",ans);

    }
    return 0;
}

 

你可能感兴趣的:(数据结构)