2090. 「ZJOI2016」旅行者 分治，最短路

2090. 「ZJOI2016」旅行者

链接

loj

思路

\((l,mid)(mid+1,r)\).考虑跨过mid的贡献。

假设选的中间那条线的点为gzy,贡献为\(dis(x,gzy)+dis(gzy,y)\)

那就计算n遍最短路,一次分治为\(n^2mlog{nm}\)

设S=n*m.矩阵的长度是不定的，每次取最长的边进行分治是最好的，n最坏为\(\sqrt{n}\)。

\(f(n)=2*f(\frac{n}{2})+S\sqrt{S}logS。所以总的复杂度就是\)\(S\sqrt{S}logS\)

都在同侧的也需要跨一跨
代码

#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
const int _=1e5+7,INF=0x3f3f3f3f;
int read() {int x=0,f=1;char s=getchar();for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';return x*f;
}
int n,m,q,ans[_],vis[_];
struct node {int x,y,X,Y,u,v,id;
}Q[_],tmp[_];
bool cmp(node a,node b) {return a.id<b.id;}
struct edge {int v,nxt,q;}e[_];
int head[_],tot;
void add(int u,int v,int q) {e[++tot].v=v;e[tot].q=q;e[tot].nxt=head[u];head[u]=tot;
}
int id(int x,int y) {return (x-1)*m+y;}
struct T_T {int u,val;T_T(int a=0,int b=0) {u=a,val=b;}bool operator < (const T_T &b) const {return val>b.val;}
};
int dis[_];
void dij(int S) {dis[S]=0;priority_queue<T_T> q;q.push(T_T(S,0));while(!q.empty()) {T_T u=q.top();q.pop();if(dis[u.u]!=u.val) continue;for(int i=head[u.u];i;i=e[i].nxt) {int v=e[i].v;if(vis[v]&&dis[v]>u.val+e[i].q) {dis[v]=u.val+e[i].q;q.push(T_T(v,dis[v]));}}}
}
void solve(int x,int y,int X,int Y,int l,int r) {if(l>r) return;if(x==X&&y==Y) {for(int i=l;i<=r;++i) ans[Q[i].id]=0;return;}if(Y-y>X-x) {int mid=(Y+y)>>1;FOR(i,x,X) {FOR(j,x,X) FOR(k,y,Y) dis[id(j,k)]=INF;dij(id(i,mid));FOR(j,l,r)ans[Q[j].id]=min(dis[Q[j].u]+dis[Q[j].v],ans[Q[j].id]);}FOR(i,x,X) vis[id(i,mid)]=0;int p=l,q=r;FOR(i,l,r) {if(max(Q[i].y,Q[i].Y)<=mid) tmp[p++]=Q[i];if(min(Q[i].y,Q[i].Y)>mid) tmp[q--]=Q[i];}FOR(i,l,r) Q[i]=tmp[i];solve(x,y,X,mid,l,p-1);solve(x,mid+1,X,Y,q+1,r);} else {int mid=(X+x)>>1;FOR(i,y,Y) {FOR(j,x,X) FOR(k,y,Y) dis[id(j,k)]=INF;dij(id(mid,i));FOR(j,l,r)ans[Q[j].id]=min(dis[Q[j].u]+dis[Q[j].v],ans[Q[j].id]);}FOR(i,y,Y) vis[id(mid,i)]=0;int p=l,q=r;FOR(i,l,r) {if(max(Q[i].x,Q[i].X)<=mid) tmp[p++]=Q[i];if(min(Q[i].x,Q[i].X)>mid) tmp[q--]=Q[i];}FOR(i,l,r) Q[i]=tmp[i];solve(x,y,mid,Y,l,p-1);solve(mid+1,y,X,Y,q+1,r);}
}
int main() {n=read(),m=read();FOR(i,1,n) FOR(j,1,m-1) {int val=read();add(id(i,j),id(i,j+1),val);add(id(i,j+1),id(i,j),val);}FOR(i,1,n-1) FOR(j,1,m) {int val=read();add(id(i,j),id(i+1,j),val);add(id(i+1,j),id(i,j),val);}FOR(i,1,n) FOR(j,1,m) vis[id(i,j)]=1;q=read();FOR(i,1,q) {Q[i].x=read(),Q[i].y=read(),Q[i].u=id(Q[i].x,Q[i].y);Q[i].X=read(),Q[i].Y=read(),Q[i].v=id(Q[i].X,Q[i].Y);Q[i].id=i,ans[i]=INF;}solve(1,1,n,m,1,q);FOR(i,1,q) printf("%d\n",ans[i]);return 0;
}