4456: [Zjoi2016]旅行者 分治+dijkstra

Description
小Y来到了一个新的城市旅行。她发现了这个城市的布局是网格状的，也就是有n条从东到西的道路和m条从南到北的道路，这些道路两两相交形成n×m个路口 (i,j)(1≤i≤n,1≤j≤m)。她发现不同的道路路况不同，所以通过不同的路口需要不同的时间。通过调查发现，从路口(i,j)到路口(i,j+1)需要时间 r(i,j)，从路口(i,j)到路口(i+1,j)需要时间c(i,j)。注意这里的道路是双向的。小Y有q个询问，她想知道从路口(x1,y1)到路口(x2,y2)最少需要花多少时间。

题解：

一开始看错数据范围了……以为是$n,m$都是$2\times10^4$……其实知道这是分治之后还是很简单的，就直接让中线上的点跑到每个点的最短路就行了，对于一些两个点都在左或右区间的询问再递归下去求解。dijkstra重载符号的时候反了看了好久才看出来…

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=20010;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,m,v[Maxn][4];
int p(int x,int y){return (x-1)*m+y;}
struct Query{int x1,y1,x2,y2,id;}q[100010],ql[100010],qr[100010];int Q;
int ans[100010];
struct Node
{
    int dis,x,y;
    Node(int _dis,int _x,int _y){dis=_dis,x=_x,y=_y;}
};
bool operator > (Node a,Node b){return a.dis>b.dis;}
int tx[]={-1,0,1,0};
int ty[]={0,1,0,-1};
int f[Maxn];bool vis[Maxn];
void dijkstra(int stx,int sty,int lx,int rx,int ly,int ry)
{
    for(int i=lx;i<=rx;i++)
    for(int j=ly;j<=ry;j++)
    f[p(i,j)]=inf,vis[p(i,j)]=false;f[p(stx,sty)]=0;
    priority_queue<Node,vector<Node>,greater<Node> >Queue;
    Queue.push(Node(0,stx,sty));
    while(!Queue.empty())
    {
        Node t=Queue.top();Queue.pop();
        int x=t.x,y=t.y;
        if(vis[p(x,y)])continue;
        vis[p(x,y)]=true;
        for(int i=0;i<4;i++)
        {
            int nx=x+tx[i],ny=y+ty[i];
            if(nx>=lx&&nx<=rx&&ny>=ly&&ny<=ry&&f[p(x,y)]+v[p(x,y)][i]<f[p(nx,ny)])
            {
                f[p(nx,ny)]=f[p(x,y)]+v[p(x,y)][i];
                Queue.push(Node(f[p(nx,ny)],nx,ny));
            }
        }
    }
}
void solve(int lx,int rx,int ly,int ry,int lq,int rq)
{
    if(lq>rq)return;
    if(lx==rx&&ly==ry){for(int i=lq;i<=rq;i++)ans[q[i].id]=0;return;}
    if(rx-lx>ry-ly)
    {
        int mid=lx+rx>>1,ll=0,lr=0;
        for(int i=ly;i<=ry;i++)
        {
            dijkstra(mid,i,lx,rx,ly,ry);
            for(int j=lq;j<=rq;j++)
            ans[q[j].id]=min(ans[q[j].id],f[p(q[j].x1,q[j].y1)]+f[p(q[j].x2,q[j].y2)]);
        }
        for(int i=lq;i<=rq;i++)
        {
            if(q[i].x1<=mid&&q[i].x2<=mid)ql[++ll]=q[i];
            if(q[i].x1>mid&&q[i].x2>mid)qr[++lr]=q[i];
        }
        for(int i=lq;i<=lq+ll-1;i++)q[i]=ql[i-lq+1];
        for(int i=lq+ll;i<=lq+ll+lr-1;i++)q[i]=qr[i-lq-ll+1];
        solve(lx,mid,ly,ry,lq,lq+ll-1),solve(mid+1,rx,ly,ry,lq+ll,lq+ll+lr-1);
    }
    else
    {
        int mid=ly+ry>>1,ll=0,lr=0;
        for(int i=lx;i<=rx;i++)
        {
            dijkstra(i,mid,lx,rx,ly,ry);
            for(int j=lq;j<=rq;j++)
            ans[q[j].id]=min(ans[q[j].id],f[p(q[j].x1,q[j].y1)]+f[p(q[j].x2,q[j].y2)]);
        }
        for(int i=lq;i<=rq;i++)
        {
            if(q[i].y1<=mid&&q[i].y2<=mid)ql[++ll]=q[i];
            if(q[i].y1>mid&&q[i].y2>mid)qr[++lr]=q[i];
        }
        for(int i=lq;i<=lq+ll-1;i++)q[i]=ql[i-lq+1];
        for(int i=lq+ll;i<=lq+ll+lr-1;i++)q[i]=qr[i-lq-ll+1];
        solve(lx,rx,ly,mid,lq,lq+ll-1),solve(lx,rx,mid+1,ry,lq+ll,lq+ll+lr-1);
    }
}
int main()
{
    memset(ans,127,sizeof(ans));
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    for(int j=1;j<m;j++)
    v[p(i,j)][1]=v[p(i,j+1)][3]=read();
    for(int i=1;i<n;i++)
    for(int j=1;j<=m;j++)
    v[p(i,j)][2]=v[p(i+1,j)][0]=read();
    Q=read();
    for(int i=1;i<=Q;i++)q[i].x1=read(),q[i].y1=read(),q[i].x2=read(),q[i].y2=read(),q[i].id=i;
    solve(1,n,1,m,1,Q);
    for(int i=1;i<=Q;i++)printf("%d\n",ans[i]);
}