标签：初始   ==   lib   algorithm   char   序列   efi   tchar   set   

题意

你有一支由 n 名预备役士兵组成的部队，士兵从 1 到 n 编号，要将他们拆分 成若干特别行动队调入战场。出于默契的考虑，同一支特别行动队中队员的编号 应该连续，即为形如(i, i + 1, ..., i + k)的序列。 编号为 i 的士兵的初始战斗力为 xi ，一支特别行动队的初始战斗力 x 为队内 士兵初始战斗力之和，即$ x = x_i + x_{i+1} + ... + x_{i+k}$。

通过长期的观察，你总结出一支特别行动队的初始战斗力 x 将按如下经验公 式修正为 \(x':x'= ax^2+bx+cx\) ，其中 a, b, c 是已知的系数（a < 0）。 作为部队统帅，现在你要为这支部队进行编队，使得所有特别行动队修正后 战斗力之和最大。试求出这个最大和。

例如，你有 4 名士兵，$ x_1 = 2, x_2 = 2, x_3 = 3, x_4 = 4$经验公式中的参数为 a = –1, b = 10, c = –20。此时，最佳方案是将士兵组成 3 个特别行动队：第一队包含士兵 1 和士兵 2，第二队包含士兵 3，第三队包含士兵 4。特别行动队的初始战斗力分 别为 4, 3, 4，修正后的战斗力分别为 4, 1, 4。修正后的战斗力和为 9，没有其它 方案能使修正后的战斗力和更大。

\(n \leq 10^6\)

分析

考虑dp，容易列出dp方程
\[ dp[i]=\max\{ dp[j] +a(s[i]-s[j])^2 + b(s[i] - s[j]) +c\} (j<i) \]
\(dp[i]\)表示前\(i\)个的最大修正战斗力，\(s[i]\)是初始战斗力的前缀和。

显然是\(O(n^2)\)的，观察式子，发现可以变形做斜率优化。

令\(j>k\)（这个很重要，保证了除式大于0，至于为什么要这样，那是尝试得出来的），假设\(j\)比\(k\)优，得到
\[ dp[j] +as[j]^2 -bs[j] - (dp[k] + as[k]^2 -bs[k]) > 2as[i](s[j]-s[k]) \]
然后这里由于\(a<0\)，除过去不利于分析，所以保留在右方，这时\(j > k\)的作用体现出来了，\(s[j] > s[k]\)保证除过去不变号，且横坐标递增。
\[ \frac{dp[j] +as[j]^2 -bs[j] - (dp[k] + as[k]^2 -bs[k])}{s[j]-s[k]} > 2as[i] \]
然后是\(>\)号，且横坐标\(s\)递增，所以维护上凸包。

由于\(a<0\)，所以\(2as\)递减，所以可以用单调队列。

简单来说，能用单调队列的是

1. 大于单调减
2. 小于单调增

时间复杂度\(O(n)\)

代码

没写return，-Wall又不知道怎么自动关了，调得很难受。感谢W学姐帮我看出来。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<algorithm>
#include<bitset>
#include<cassert>
#include<ctime>
#include<cstring>
#define rg register
#define il inline
#define co const
template<class T>il T read()
{
    rg T data=0;
    rg int w=1;
    rg char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch=='-')
            w=-1;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        data=data*10+ch-'0';
        ch=getchar();
    }
    return data*w;
}
template<class T>T read(T&x)
{
    return x=read<T>();
}
using namespace std;
typedef long long ll;

co int N=1e6+2;
ll a,b,c;
ll x[N];
ll dp[N];

ll Up(int j,int k)
{
    return dp[j]+a*x[j]*x[j]-b*x[j]-dp[k]-a*x[k]*x[k]+b*x[k];
}

ll Down(int j,int k)
{
    return x[j]-x[k];
}

ll Cal(int i,int j)
{
    return dp[j]+a*(x[i]-x[j])*(x[i]-x[j])+b*(x[i]-x[j])+c; // edit 1: return
}

int q[N];

int main()
{
//  freopen(".in","r",stdin);
//  freopen(".out","w",stdout);
    int n=read<int>();
    read(a),read(b),read(c);
    for(int i=1;i<=n;++i)
        x[i]=x[i-1]+read<int>();
    int head=0,tail=0;
    q[tail++]=0;
    for(int i=1;i<=n;++i)
    {
        while(head+1<tail&&Up(q[head+1],q[head])>=2*a*x[i]*Down(q[head+1],q[head]))
            ++head;
        dp[i]=Cal(i,q[head]);
        while(head+1<tail&&Up(i,q[tail-1])*Down(q[tail-1],q[tail-2])>=Up(q[tail-1],q[tail-2])*Down(i,q[tail-1]))
            --tail;
        q[tail++]=i;
    }
    printf("%lld\n",dp[n]);
    return 0;
}

LG3628 [APIO2010]特别行动队

标签：初始   ==   lib   algorithm   char   序列   efi   tchar   set   

原文地址：https://www.cnblogs.com/autoint/p/10192446.html