有n个有先后顺序的句子,每个句子有一个长度\(a_i\),现在要求将其划分成段,两个句子间会存在一个空格,每一段的权值为|总长度-L|的p次方,总权值为所有的段权值之和,询问总权值的最小值,并输出方案,如果答案超过\(10^{18}\),输出Too hard to arrange,\(n\leq 100000,l\leq 3000000,p\leq 10\),每个句子长度不超过30.
解
显然要表现处理到哪一个句子,于是设\(f_i\)表示处理到第i个句子的最小总权值,\(s_i\)为前i个句子的长度前缀和,有
\[f_i=\min_{j=0}^{i-1}\{f_j+|s_i-s_j+i-j-1-l|^p\}\]
容易知道,这是\(O(n^2)\)递推,而且高次不符合斜率优化和单调栈优化,于是考虑四边形不等式,设\(w(j,i)=|s_i-s_j+i-j-1-l|^p\),现在要证明
\(w(j,i+1)+w(j+1,i)\geq w(j,i)+w(j+1,i+1)\)
即证
\(w(j,i+1)-w(j,i)\geq w(j+1,i+1)-w(j+1,i)\)
设\(A=s_i-s_j+i-j-1-l,B=s_i-s_{j+1}+i-j-2-l\),显然\(A>B\),于是即证
\(|A+a_i+1|^p-|A|^p\geq |B+a_i+1|^p-|B|^p\)
由于函数\(|x+c|^p\)单调递增,故成立,于是w满足四边形不等式。
参考代码:
#include <iostream>
#include <cstdio>
#define il inline
#define ri register
#define ll long double
#define Size 100010
using namespace std;
struct inter{
int l,r,j;
}T[Size];
ll dp[Size];
string s[Size];
int sum[Size],pre[Size]
,l,p,A,B;
void print(int);
il void work();
il int dfs(inter&,int);
il ll pow(ll,int),tran(int,int);
template<class free>il free Abs(free);
int main(){
int lsy;scanf("%d",&lsy);
while(lsy--)work(),puts("--------------------");
return 0;
}
void print(int x){
if(x){
print(pre[x]);
for(int i(pre[x]+1);i<x;++i)
cout<<s[i]<<' ';cout<<s[x]<<endl;
}
}
il int dfs(inter&a,int x){
int l(a.l),r(a.r),mid;
while(l<=r){
mid=l+r>>1;
if(tran(mid,a.j)<tran(mid,x))l=mid+1;
else r=mid-1;
}return l;
}
template<class free>
il free Abs(free x){
return x<0?-x:x;
}
il ll pow(ll x,int y){
ll ans(1);while(y){
if(y&1)ans*=x;
x*=x,y>>=1;
}return ans;
}
il ll tran(int i,int j){
return dp[j]+pow(Abs(sum[i]-sum[j]+i-j-1-l),p);
}
il void work(){
int n;
scanf("%d%d%d",&n,&l,&p);
for(int i(1);i<=n;++i)
cin>>s[i],sum[i]=sum[i-1]+s[i].size();
A=B=1,T[1]={1,n,0};
for(int i(1),j;i<=n;++i){
while(A<=B&&T[A].r<i)++A;dp[i]=tran(i,T[A].j),pre[i]=T[A].j,++T[A].l;
while(A<=B&&tran(T[B].l,T[B].j)>=tran(T[B].l,i))j=T[B].l,--B;
if(A<=B)j=dfs(T[B],i),T[B].r=j-1;if(j<=n)T[++B].l=j,T[B].r=n,T[B].j=i;
}
if(dp[n]>1e18)puts("Too hard to arrange");
else printf("%lld\n",(long long)dp[n]),print(n);
}