HDU.2829.Lawrence(DP 斜率優化)
阿新 • • 發佈:2018-02-07
digi pos == spa tchar str 轉移 方程 決策單調
題目鏈接
\(Description\)
給定一個\(n\)個數的序列,最多將序列分為\(m+1\)段,每段的價值是這段中所有數兩兩相乘的和。求最小總價值。
\(Solution\)
寫到這突然懶得寫了。。
丟個題解走人
/*
樸素O(n^3):f[i][j]表示當前在i分了j段的最小價值 W[i]表示1~i的總價值 S[i]表示1~i的原序列值之和
則有 f[i][j]=min{ f[k][j-1]+W[i]-W[k]-S[i]*(S[i]-S[k]) } (1≤k<i)
這個方程可以用斜率優化 不過好像首先有個決策單調
*/
#include<cstdio>
#include<cctype>
#define gc() getchar()
typedef long long LL;
const int N=1e3+5;
int n,m,A[N],q[N];
LL W[N],S[N],f[N][2];
inline int read()
{
int now=0,f=1;register char c=gc();
for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now*f;
}
inline LL Calc_X(int x,int y)
{
return S[x]-S[y];
}
inline LL Calc_Y(int x,int y,int s)
{
return (f[x][s]+S[x]*S[x]-W[x])-(f[y][s]+S[y]*S[y]-W[y]);
}
inline LL Calc_f(int x,int y,int s)
{
return f[x][s]+W[y]-W[x]-S[x]*(S[y]-S[x]);
}
int main()
{
while(n=read(),m=read(),n&&m)
{
S[0]=W[0 ]=0;
for(int i=1;i<=n;++i)
A[i]=read(), S[i]=S[i-1]+A[i], W[i]=W[i-1]+A[i]*S[i-1];
for(int i=1;i<=n;++i) f[i][0]=W[i];
int p=1;
for(int h,t,j=1;j<=m;++j)//至多分m+1段
{
h=t=1, q[1]=0;//i從0轉移就是W[i]
p^=1;
for(int i=1;i<=n;++i)
{
while(h<t && Calc_Y(q[h+1],q[h],p)<S[i]*Calc_X(q[h+1],q[h]))//決策單調 所以對於當前i不會成為最優值的,之後都不會成為最優值
++h;
f[i][p^1]=Calc_f(q[h],i,p);
while(h<t && Calc_Y(i,q[t],p)*Calc_X(q[t],q[t-1]) <= Calc_Y(q[t],q[t-1],p)*Calc_X(i,q[t]))//維護隊尾下凸包性質
--t;
q[++t]=i;
}
}
printf("%lld\n",f[n][p^1]);
}
return 0;
}
HDU.2829.Lawrence(DP 斜率優化)