luogu P4360 [CEOI2004]锯木厂选址

背景：

有些累…

题目传送门：

https://www.luogu.org/problemnew/show/P4360

题意：

现在有$n$棵树，每一棵树有对应的质量$a_i$，且相邻的两棵树的距离是$b_i$（$i$与$i+1$的距离），现在在第$n$棵树的位置后有一个锯木厂，但是还要修建$2$个锯木厂。现在求运输所有树到锯木厂的最小费用。

思路：

恶心毒瘤题。

设$f_i$表示在前$i$棵树中修建了$2$个锯木厂且要在$i$修建一个锯木厂的最小费用。
设$A_k={\sum }_{i=1}^k{a}_i$。
设$B_k={\sum }_{i=1}^{k-1}{b}_i$（即$1$到$i$的距离）。
设$T_{j,i}$表示$i$棵树的位置建一个锯木厂，从$j+1$到$i$之间的树全都运往$i$。

得到状态转移方程式：
$f_i={\min }_{j=1}^{i-1}{T}_{0,j}+T_{j,i}+T_{i,n+1}$。
这里解释一下：$0+1=1$到$j$棵树运往$j$厂；$j+1$到$i$棵树运往$i$厂；$i+1$到$n$棵树运往最后一个厂$n+1$，由于$n+1$位置上没有树，所以记作$T_{i,n+1}$会更方便且不影响结果。

考虑$T_{j,i}$的表示。
$T_{j,i}={\sum }_{k=j+1}^{i-1}{a}_k\ast (B_i-B_k)$
开始一波化简。
$T_{j,i}={\sum }_{k=j+1}^{i-1}(a_k\ast B_i-a_k\ast B_k)$
$T_{j,i}=(B_i\ast {\sum }_{k=j+1}^{i-1}{a}_k)-({\sum }_{k=j+1}^{i-1}{B}_k\ast a_k)$
设$s_i=B_i\ast a_i$，则：
$T_{j,i}=B_i\ast (A_{i-1}-A_j)-({\sum }_{k=j+1}^{i-1}{s}_k)$
设$S_k={\sum }_{i=1}^n{s}_i$，则：
$T_{j,i}=B_i\ast (A_{i-1}-A_j)-(S_{i-1}-S_j)$
此时时间复杂度变为$\Theta (n^2)$。

带回得：$f_i=(A_{j-1}{B}_j-S_{j-1})+(B_i(A_{i-1}-A_j)-(S_{i-1}-S_j))+(B_{n+1}(A_{n+1-1}-A_i)-(S_{n+1-1}-S_i))$
$f_i=(A_{j-1}{B}_i-S_{j-1})+(A_{i-1}{B}_i-A_j{B}_i-S_{i-1}+S_j)+(A_n{B}_{n+1}-A_i{B}_{n+1}-S_n+S_i)$
设$T{0}_i=A_{i-1}{B}_i-S_{i-1}$，显然可以被预处理出来，则：
$f_i=T{0}_j+(A_{i-1}{B}_i-A_j{B}_i-S_{i-1}+S_j)+(A_n{B}_{n+1}-A_i{B}_{n+1}-S_n+S_i)$
得到斜率方程为：$\frac{T_{0,j}-T_{0,k}+S_j-S_k}{A_j-A_k}<B_i$。
就$OK$了。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
	int n;
	LL ans=(LL)(2e10);
	LL a[100010],A[100010],b[100010],B[100010],s[100010],S[100010],T0[100010],f[100010];
	int que[100010];
LL calc1(int x,int y)
{
	return A[y]-A[x];
}
LL calc2(int x,int y)
{
	return (T0[y]+S[y])-(T0[x]+S[x]);
}
double calc(int x,int y)
{
	return (double)(calc2(x,y))/(double)(calc1(x,y));
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%lld %lld",&a[i],&b[i]);
	for(int i=1;i<=n+1;i++)
	{
		A[i]=A[i-1]+a[i];
		B[i]=B[i-1]+b[i-1];
		s[i]=a[i]*B[i];
		S[i]=S[i-1]+s[i];
		T0[i]=A[i-1]*B[i]-S[i-1];
	}
	int head=1,tail=1;
	for(int i=1;i<=n;i++)
	{
		while(head<tail&&calc(que[head],que[head+1])<=(double)B[i]) head++;
		f[i]=T0[que[head]]+(A[i-1]*B[i]-A[que[head]]*B[i]-S[i-1]+S[que[head]])+(A[n]*B[n+1]-A[i]*B[n+1]-S[n]+S[i]);
		ans=min(ans,f[i]);
		while(head<tail&&calc(que[tail],i)<=calc(que[tail-1],que[tail])) tail--;
		que[++tail]=i;
	}
	printf("%lld",ans);
}