传送门
解析:
优秀的斜率优化DP。
首先DP式子不是很好想,再加上本蒟蒻很久没写斜率优化,这道题就场上直接咕咕咕了。。。
思路:
不要想一次四个方向处理完(不然下场和我一样,无法处理后效性),我们可以处理四次,分别处理左上右下右上左下,最后所有答案取一个就可以了。
那么考虑怎么处理左上方向,其他的显然可以通过对称转变一下。
考虑我们已经处理到第列,我们可以维护前行每一行平移过来最近的。第行记为,所以我们的决策就是
这个东西可以斜率优化一下,即
考虑斜率优化,另若是较优决策则有
即
然后斜率优化乱搞就行了。
然后发现是单调的,所以可以直接来单调队列维护下凸壳就行了。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const
cs int N=1003;
int n,m;
bool way[N][N];
int ans[N][N];
int f[N][N];
int last1[N];
struct Point{
int x,y;
Point(cs int &_x=0,cs int &_y=0):x(_x),y(_y){}
friend ll operator*(cs Point &a,cs Point &b){return 1ll*a.x*b.y-1ll*a.y*b.x;}
friend Point operator-(cs Point &a,cs Point &b){
return Point(a.x-b.x,a.y-b.y);
}
}p[N];
inline void solve(){
fill(last1+1,last1+m+1,N<<1);
for(int re i=1;i<=n;++i){
int head=1,tail=0;
for(int re j=1;j<=m;++j){
++last1[j];
if(way[i][j])last1[j]=0;
Point tmp=Point(j,last1[j]*last1[j]+j*j);
while(head<tail&&(p[tail]-p[tail-1])*(tmp-p[tail-1])<=0)--tail;
p[++tail]=tmp;
while(head<tail&&p[head].y-2*j*p[head].x>p[head+1].y-2*j*p[head+1].x)++head;
f[i][j]=p[head].y-2*j*p[head].x+j*j;
}
}
}
signed main(){
ios::sync_with_stdio(false);
cin>>n>>m;
for(int re i=1;i<=n;++i){
string sss;
cin>>sss;
for(int re j=1;j<=m;++j)way[i][j]=sss[j-1]^48;
}
solve();
for(int re i=1;i<=n;++i)
for(int re j=1;j<=m;++j)ans[i][j]=f[i][j];
for(int re i=1;i<=n;++i)reverse(way[i]+1,way[i]+m+1);
solve();
for(int re i=1;i<=n;++i)
for(int re j=1;j<=m;++j)ans[i][j]=min(ans[i][j],f[i][m-j+1]);
for(int re i=1;i*2<=n;++i)swap(way[i],way[n-i+1]);
//交换数组的swap和交换变量的swap不是同一个,有不同的声明和实现。
solve();
for(int re i=1;i<=n;++i)
for(int re j=1;j<=m;++j)ans[i][j]=min(ans[i][j],f[n-i+1][m-j+1]);
for(int re i=1;i<=n;++i)reverse(way[i]+1,way[i]+m+1);
solve();
for(int re i=1;i<=n;++i)
for(int re j=1;j<=m;++j)ans[i][j]=min(ans[i][j],f[n-i+1][j]);
for(int re i=1;i<=n;++i){
for(int re j=1;j<=m;++j)
printf("%d ",ans[i][j]);
pc('\n');
}
return 0;
}