【题解】P11459 [USACO24DEC] It's Mooin' Time P
其实是套路题。但有段时间没写过闵可夫斯基和优化 \(dp\) 了,记录一下。
先简单回顾一下闵可夫斯基和。点集 \(P\) 与 点集 \(Q\) 的闵可夫斯基和定义为:\(P+Q=\{a+b|a\in P,b\in Q\}\) 。
对于两个凸包的闵可夫斯基和,有如下性质:
- \(P+Q\) 也为凸包。
- \(P+Q\) 的边集由 \(P,Q\) 的边按极角排序后连接的结果。
因此对于大小为 \(n\) 和 \(m\) 的两个凸包,我们可以通过 \(O(n+m)\) 的时间复杂度求出它们的闵可夫斯基和。
对于 \((\max,+)\) 卷积和 \((\min,+)\) 卷积,我们也可以通过类似的方式进行优化。
以 \((\max,+)\) 卷积为例,设 $c_i=\max_{j+k=i}{a_j+b_k} $。
如果 \((i,a_i)\) 与 \((i,b_i)\) 分别构成凸包,它们的闵可夫斯基和即为 \((i,c_i)\) ,那么可以通过如上方法快速求解数组 \(c\) 。
如何使用闵可夫斯基和优化 \(dp\) ?对于 \(f_{i,j}=\max_{k<j}\{f_{i-1,k}+a_i\}\) 这类的 \(dp\) ,其中 \(i\) 这一维可能是形如 “选择 \(i\) 个物品” 状物。我们可以退一步,将序列 \(dp\) 改为区间 \(dp\) ,然后用 \([l,mid]\) 和 \([mid+1,r]\) 的答案合并得到 \([l,r]\) 的答案。考虑分治,这样合并两段区间的贡献相当于做闵可夫斯基和的过程,总复杂度 \(O(n\log n)\) 。
回到这一题,由于 \(L\) 固定,我们可以对每个 \(i\) 预先求出 \([i,i+L-1]\) 合并出 MO.. 的答案。设 \(f_{i,j}\) 表示考虑前 \(i\) 位作为 MO.. 的起点,恰好包含 \(j\) 段 MO.. 的答案,那么转移形式与上式类似,只是要求前一个转移点与当前点相差至少 \(L\) 。将 \(dp\) 改为区间 \(dp\) ,设 \(f_{l,r,a1,a2,k}\) 表示考虑第 \(l\) 到 \(r\) 位作为 MO.. 的起点,最左、右端的点离左右端点的距离状态分别为 \(a_1,a_2\) ,至少包含 \(k\) 段 MO.. 的答案。其中,\(a_1,a_2\) 为 \(0/1/2\) 分别表示距离两段距离为 \(0/1/\geq2\) 。合并连段时做类似矩阵乘法,对 \(a_1,a_2\) 做卷积。时间复杂度 \(O(nL^2\log n)\) 。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
inline ll read(){
ll x=0; bool f=1; char ch=getchar();
while(ch<'0'||ch>'9'){
if(ch=='-') f=0; ch=getchar();
}
while(ch>='0'&&ch<='9'){
x=(x<<1)+(x<<3)+(ch^48); ch=getchar();
}
return f?x:-x;
}
inline void write(ll x){
if(x<0) x=-x,putchar('-');
if(x>9) write(x/10);
putchar('0'+x%10);
}
const ll N=300009;
const ll INF=1e18;
ll L,n,c[N],a[N];
ll g[10][10];
string s;
struct Node{
vector<ll> f[3][3];
};
vector<ll> merge(vector<ll> a,vector<ll> b){
for(ll i=a.size()-1;i>=2;i--) a[i]-=a[i-1];
for(ll i=b.size()-1;i>=2;i--) b[i]-=b[i-1];
vector<ll> c(a.size()+b.size()-1);
c[0]=a[0]+b[0];
merge(a.begin()+1,a.end(),b.begin()+1,b.end(),c.begin()+1);
for(ll i=1;i<a.size()+b.size()-1;i++){
c[i]+=c[i-1];
}
return c;
}
void add(vector<ll>&x,vector<ll> y){
while(x.size()<y.size()) x.push_back(INF);
for(ll i=0;i<y.size();i++) x[i]=min(x[i],y[i]);
}
Node solve(ll l,ll r){
if(r-l+1<=2*L){
Node res;
for(ll x=0;x<L;x++){
for(ll y=0;y<L;y++){
memset(g,0x3f,sizeof(g));
for(ll i=l+x;i<=r-y;i++){
g[i-l+1][1]=a[i];
if(i-L>=l){
for(ll j=2;j<=r-l+1;j++){
g[i-l+1][j]=min(g[i-l+1][j],g[i-l+1-L][j-1]+a[i]);
}
}
for(ll j=1;j<=r-l+1;j++){
g[i-l+1][j]=min(g[i-l+1][j],g[i-l][j]);
}
}
vector<ll> cur; cur.push_back(0);
for(ll i=1;i<=r-l+1;i++){
ll mn=INF;
for(ll j=l;j<=r;j++){
mn=min(mn,g[j-l+1][i]);
}
if(mn>=INF) break;
cur.push_back(mn);
}
res.f[x][y]=cur;
}
}
return res;
}
ll mid=(l+r)/2;
auto ls=solve(l,mid),rs=solve(mid+1,r);
Node res;
for(ll i=0;i<=L-1;i++){
for(ll j=0;j<=L-1;j++){
for(ll mid=0;mid<=L-1;mid++){
add(res.f[i][j],merge(ls.f[i][mid],rs.f[L-1-mid][j]));
}
}
}
return res;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
cin>>L>>n;
cin>>s; s=" "+s;
for(ll i=1;i<=n;i++){
cin>>c[i];
}
for(ll i=1;i<=n-L+1;i++){
if(s[i]!='M') a[i]=c[i];
for(ll j=i+1;j<=i+L-1;j++){
if(s[j]!='O') a[i]+=c[j];
}
}
Node res=solve(1,n-L+1);
for(ll i=1;i<=n/L;i++){
cout<<res.f[0][0][i]<<"\n";
}
return 0;
}

浙公网安备 33010602011771号