蓝桥杯官网 试题 PREV-236 历届真题 出租车【第十一届】【决赛】【研究生组】【C++】一种解法

#include<cstdio>
#include<cstring>
#include<algorithm>
const int maxn=40010,maxm=40010;
const int inf=0x3fffffff;
int mn,jj,uu,n,m,num,len,next[maxm],ev[maxm],memo,idu,idv;
int temp[101],nbs[maxn],ps[maxn],heap[maxn],horpa[10001],verpa[10001],tot[10001];
bool mk[maxn];
double value[maxn],ver[101],hor[101],ew[maxm],wt,ans,idw;
double md(double a,int b)
{
    int i=int(a/b+0.0000001);
    return a-i*b;
}
void update(int r)
{
    int q=ps[r],p=q>>1;
    while(p&&value[heap[p]]>value[r])
    {
        ps[heap[p]]=q;
        heap[q]=heap[p];
        q=p;
        p=q>>1;
    }
    heap[q]=r;
    ps[r]=q;
}
int getmin()
{ 
    int ret=heap[1],p=1,q=2,r=heap[len--];
    while(q<=len)
    {
        if(q<len&&value[heap[q+1]]<value[heap[q]])q++;
        if(value[heap[q]]<value[r])
        {
            ps[heap[q]]=p;
            heap[p]=heap[q];
            p=q;
            q=p<<1;
        }
        else break;
    }
    heap[p]=r;
    ps[r]=p;
    return ret;
}
void dijkstra(int src,int dst)
{ 
    int i,j,u,v;
    std::fill(value,value+mn+5,inf);
    std::fill(mk,mk+mn+5,false);
    std::fill(ps,ps+mn+5,0);
    value[src]=0;
    heap[len=1]=src;
    ps[src]=1;
    while(!mk[dst])
    {
        if(len==0)return;
        u=getmin();
        if(u==dst)return;
        mk[u]=true;
        for(j=nbs[u];j;j=next[j])
        {
            v=ev[j];
            if(!mk[v]&&value[u]+ew[j]<value[v])
            {
                if(ps[v]==0)
                {
                    heap[++len]=v;
                    ps[v]=len;
                }
                value[v]=value[u]+ew[j];
                update(v);
            }
        }
        if(u!=src)
        for(jj=0;jj<4;jj++)//check around
        {
            uu=u/4*4+jj;
            if(uu+1==u||uu==u+3||uu==u)
            {
                for(j=nbs[uu];j;j=next[j])
                {
                    v=ev[j];
                    if(!mk[v]&&value[u]+ew[j]<value[v])
                    {
                        if(ps[v]==0)
                        {
                            heap[++len]=v;
                            ps[v]=len;
                        }
                        value[v]=value[u]+ew[j];
                        update(v);
                    }
                }
            }
            else
            {
                if(md(ans+value[u],tot[u/4])<verpa[u/4])//vertical is green
                {
                    if(u%2)wt=0;
                    else wt=verpa[u/4]-md(ans+value[u],tot[u/4]);
                }
                else
                {
                    if(u%2==0)wt=0;
                    else wt=tot[u/4]-md(ans+value[u],tot[u/4]);
                }
                for(j=nbs[uu];j;j=next[j])
                {
                    v=ev[j];
                    if(!mk[v]&&value[u]+wt+ew[j]<value[v])
                    {
                        if(ps[v]==0)
                        {
                            heap[++len]=v;
                            ps[v]=len;
                        }
                        value[v]=value[u]+wt+ew[j];
                        update(v);
                    }
                }
            }
        }
    }
}
int di[4]={0,1,0,-1},dj[4]={1,0,-1,0};
void getid(int ii1,int jj1,int ii2,int jj2)
{
    if(ii1!=ii2)//vertical
    {
        if(ii1<ii2)//down
        {
            idu=(ii1*m+jj1)*4+3;
            idv=(ii2*m+jj2)*4+1;
            idw=hor[ii2];
        }
        else//up
        {
            idu=(ii1*m+jj1)*4+1;
            idv=(ii2*m+jj2)*4+3;
            idw=hor[ii1];
        }
    }
    else
    {
        if(jj1<jj2)//right
        {
            idu=(ii1*m+jj1)*4+2;
            idv=(ii2*m+jj2)*4;
            idw=ver[jj2];
        }
        else//left
        {
            idu=(ii1*m+jj1)*4;
            idv=(ii2*m+jj2)*4+2;
            idw=ver[jj1];
        }
    }
}
void addedge(int u,int v,double w)
{
    next[++num]=nbs[u];
    nbs[u]=num;
    ev[num]=v;
    ew[num]=w;
}
int main()
{
    int i,j,u,v,uu,vv,w,ii1,ii2,jj1,jj2,i1,i2,j1,j2,id,T;
    scanf("%d%d",&n,&m);
    mn=m*n*4;
    for(i=1;i<n;i++)
    {
        scanf("%d",&temp[i]);
        hor[i]=temp[i]-temp[i-1];
    }
    for(i=1;i<m;i++)
    {
        scanf("%d",&temp[i]);
        ver[i]=temp[i]-temp[i-1];
    }
    for(i=0;i<n;i++)
        for(j=0;j<m;j++)
            scanf("%d",&verpa[i*m+j]);
    for(i=0;i<n;i++)
        for(j=0;j<m;j++)
        {
            scanf("%d",&horpa[i*m+j]);
            tot[i*m+j]=verpa[i*m+j]+horpa[i*m+j];
        }
    for(ans=i1=0;i1<n;i1++)
        for(j1=0;j1<m;j1++)
            for(id=0;id<4;id++)
            {
                i2=i1+di[id];
                j2=j1+dj[id];
                if(i2>=0&&i2<n&&j2>=0&&j2<m)
                {
                       getid(i1,j1,i2,j2);
                    addedge(idu,idv,idw);
                }
            }
    scanf("%d%d%d%d",&ii1,&jj1,&ii2,&jj2);
    getid(ii1-1,jj1-1,ii2-1,jj2-1);
    idu=mn;
    addedge(idu,idv,idw/2);
    scanf("%d",&T);
    T*=2;
    while(T--)
    {
        scanf("%d%d%d%d",&i1,&j1,&i2,&j2);//jie
        getid(i1-1,j1-1,i2-1,j2-1);
        idv=mn+4;
        addedge(idu,idv,idw/2);
        dijkstra(mn,mn+4);
        nbs[idu]=next[num];num-=2;nbs[mn]=0;
        ans+=value[mn+4];
        getid(i1-1,j1-1,i2-1,j2-1);
        idu=mn;
        addedge(idu,idv,idw/2);
    }
    getid(ii1-1,jj1-1,ii2-1,jj2-1);
    idv=mn+4;
    addedge(idu,idv,idw/2);
    dijkstra(mn,mn+4);
    ans+=value[mn+4];
    printf("%.1f\n",ans);
    return 0;
}