【网络流24题----15】汽车加油行驶问题

---

喜闻乐见的分层图最短路，注意到了加油站是强制要加满油的

---

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
#define maxn 110
#define SIZE 1000000
#define llg long long 
#define inf (llg)1e16
#define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);

llg n,m,k,A,B,C,dis[maxn*maxn][15],bj[maxn*maxn][maxn],se[maxn*maxn],head,tail;
llg p[5][4];
vector<llg>a[maxn*maxn],val[maxn*maxn];

llg num(llg x,llg y){return (x-1)*n+y;}

void link(llg x,llg y,llg z){a[x].push_back(y),val[x].push_back(z);}

struct node
{
    llg x,y;
}dl[SIZE+5];

void init()
{
    p[1][0]=1,p[2][1]=1,p[3][0]=-1,p[4][1]=-1;
    cin>>n>>k>>A>>B>>C;
    for (llg i=1;i<=n*n;i++) scanf("%lld",&se[i]);
    for (llg i=1;i<=n;i++)
        for (llg j=1;j<=n;j++)
            for (llg w=1;w<=4;w++)
            {
                llg x=i+p[w][0],y=j+p[w][1];
                if (x>n || x<1 || y>n || y<1) continue;
                if (w<=2) link(num(i,j),num(x,y),0);
                else link(num(i,j),num(x,y),B);
            }
    for (llg i=1;i<=n*n;i++) for (llg j=0;j<=k;j++) dis[i][j]=inf;
    tail=1,head=0;
    dl[1].x=1;dl[1].y=k; dis[1][k]=0;
}

void spfa()
{
    llg x,v,y,w;
    do
    {
        head%=SIZE; head++;
        x=dl[head].x,y=dl[head].y;
        w=a[x].size(); bj[x][y]=0;
        if  (se[x] && y!=k)
        {
            if (dis[x][y]+A<dis[x][k])
            {
                dis[x][k]=dis[x][y]+A;
                if (!bj[x][k])
                {
                    bj[x][k]=1;
                    tail%=SIZE; tail++;
                    dl[tail].x=x,dl[tail].y=k;
                }
            }
            continue;
        }
        if (y!=0)
        {
            for (llg i=0;i<w;i++)
            {
                v=a[x][i];
                if (dis[v][y-1]>dis[x][y]+val[x][i])
                {
                    dis[v][y-1]=dis[x][y]+val[x][i];
                    if (!bj[v][y-1])
                    {
                        bj[v][y-1]=1;
                        tail%=SIZE; tail++;
                        dl[tail].x=v;dl[tail].y=y-1;
                    }
                }
            }
        }
        if (y!=k && se[x]==0)
        {
            if (dis[x][k]>dis[x][y]+C+A)
            {
                dis[x][k]=dis[x][y]+C+A;
                if (!bj[x][k])
                {
                    bj[x][k]=1;
                    tail%=SIZE; tail++;
                    dl[tail].x=x,dl[tail].y=k;
                }
            }
        }
    }while(head!=tail);
}

int main()
{
    yyj("car");
    init();
//    k--;
    spfa();
    llg ans=inf;
    for (llg i=0;i<=k;i++) ans=min(ans,dis[n*n][i]);
    cout<<ans;
    return 0;
}