[斜率优化][JZOJ5355]保命

题目描述

为了加快社会主义现代化，建设新农村，农夫约(Farmer Jo)决定给农庄做一些防火措施，保障自己、猫、奶牛的生命安全。
农夫约的农庄里有N+1 座建筑，排成了一排，编号为0～N。对于0 <=i < N，建筑i 有w[i]头奶牛居住，与建筑i+1 距离为d[i]。建筑N 已装有消防栓，现在，农夫约决定再给k 个建筑安装消防栓，以减小安全隐患。
当火灾来临时，所有奶牛会从所在建筑开始，向大编号方向逃生，直到遇上第一个消防栓（如果本来就在消防栓处，就不用跑了）。农夫约定义了一个隐患值val：所有奶牛逃生距离之和。
农夫约希望让隐患值尽可能小，需要你给他设计一个好方案。

分析

本来出题人出的是近似算法，但是出烂了，可以用dp做。
我们先列出朴素方程嘛。设f[i][j]表示在第i个位置放了第j个消防栓的距离和最小值。
f[i][k]=min(f[j][k-1]+calc(j+1,i))，calc表示j+1到k的所有位置的牛到k的距离和，这个值的计算是可以O(1)的。
设s1[i]=s1[i-1]+d[i].
s2[i]=s2[i-1]+w[i];
s3[i]=s3[i-1]+s2[i-1]*d[i-1];
那么calc(x,y)=s3[y]-s3[x-1]-s2[x-1]*(s1[y-1]-s1[x-2])
仔细看，转移式里面只有一项既和y有关，又和x有关，即s1[y-1]*s2[x-1]，然后s1是递增的，就可以斜率优化了。
斜率优化一般就是你化出一个转移方程，比如上面那个，然后用单调队列维护一个决策队列，就是把所有的有用的j存到里面，队头如果比下一个对于当前点更劣，就弹掉。每次做完当前点，我们插入它，在这里表现为f[i][k-1]，然后要把队尾一些没有用的点弹掉。
什么是没有用的点呢？我们把转移方程里面只和i有关的常数项省略掉，然后剩下的东西，和i有关的数看成未知数，那么决策j就变成了一条直线，它和其他决策是会有交点的。那么当前做完了i，决策i和队尾的交点，在决策i和队尾前一个的交点之后，就证明了队尾没用，因为在任何情况下，队尾前一个或i总有一个比队尾优。（把s1[i-1]当成自变量，它不断变大，看看几条直线的分布情况）
化式子比较关键，如果是不等式，其中变号很容易弄错，因为会把负数当成正数，项也特别多，很容易漏或者下标搞错。
那么就乱搞了。
——————10.23更新
继续来总结，今天又做了一道题，吃了大亏…
注意到优化的时候，如果斜率单调，就可以线性维护，但是此时要注意插入记得要去除不合法的线段，上凸壳和下凸壳是有区别的，做题的时候一定要画图，谨慎。
思路，就是维护好直线，然后查询的时候弹掉一些，插入的时候把一定没有用的线段弹掉。注意“没有用”的条件。另外，求交点一定要注意double.

代码

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<set>
using namespace std;
#define fo(i,j,k) for(i=j;i<=k;i++)
#define fd(i,j,k) for(i=j;i>=k;i--)
typedef long long ll;
typedef double db;
const int N=1e6+5,M=22,mo=998244353;
int n,m,i,j,k,p,dl[N],l,r,g[M][N],lst,w[N],d[N],prt[N];
ll f[M][N],s1[N],s2[N],s3[N],ans,tmp;
int read()
{
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x;
}
ll calc(int i,int j)
{
    return s3[j]-s3[i-1]-s2[i-1]*(s1[j-1]-s1[max(i-2,0)]);
}
db cross(int i,int k)
{
    return (f[p][k]-s3[k]+s2[k]*s1[k-1]-(f[p][i]-s3[i]+s2[i]*s1[i-1]))*1.0/((db)s2[k]-(db)s2[i]);
}
int main()
{
    freopen("t1.in","r",stdin);
//  freopen("t1.out","w",stdout);
    scanf("%d %d",&n,&m);
    fo(i,1,n) w[i]=read(),d[i]=read();
    fo(i,1,n+1)
    {
        s1[i]=s1[i-1]+d[i];
        s2[i]=s2[i-1]+w[i];
        s3[i]=s3[i-1]+s2[i-1]*d[i-1];
    }
    fo(i,0,m) fo(j,1,n+1) f[i][j]=1e17;
    fo(j,1,n+1) f[1][j]=s3[j];
    fo(i,2,m)
    {
        p=i-1;
        l=1;r=1;
        dl[1]=0;
        fo(j,1,n+1)
        {
            //pop();
            while (l<r&&f[i-1][dl[l]]+calc(dl[l]+1,j)>f[i-1][dl[l+1]]+calc(dl[l+1]+1,j)) l++;

            f[i][j]=f[i-1][dl[l]]+calc(dl[l]+1,j);
            g[i][j]=dl[l];

            //push();
            while (l<r&&cross(dl[r-1],dl[r])>cross(dl[r],j)) r--;
            dl[++r]=j;
        }
    }
    ans=1e17;
    fo(i,0,n)
    {
        tmp=f[m][i]+calc(i+1,n+1);
        if (ans>tmp)
        {
            lst=i;
            ans=tmp;
        }
    }
    printf("%lld\n",ans);
    prt[0]=0;
    fd(i,m,1)
    {
        prt[++prt[0]]=lst-1;
        lst=g[m][lst];
        m--;
    }
    fd(i,prt[0],1) printf("%d ",prt[i]);
}