「网络流 24 题」汽车加油行驶问题 最短路

题目链接：https://loj.ac/problem/6223

 

题意：

       给你一个100*100的方格图，你现在开着一辆满油的车在坐标（1,1），你现在要开到坐标（n，n），每开过一条边，车子的油就会少1单位，一辆满油的车只能开k单位长度，一些坐标上面会有加油站，当你经过加油站的时候你必须加油（这个必须我也是醉了，强制消费啊），并且花费A元，如果你现在所在的位置没有加油站， 你可以选择花费C元建一个加油站后再花费A元加油，因为一些原因，你必须向下（x+1）和右（y+1）的方向开，否则你就要花费B元交罚单，问你开车到点（n，n）的时候，最少的花费是多少。

做法：

       明明是网络流24题却是不是用网络流做的，这题我乍一看还以为是用最小费用流做的来着，但是仔细想想又不是，流量什么的都不能很好的被控制。想想10000个点的话好像还能再乘10给每个状态一个点，一个状态[P，h]是在P这个点，车子还有油h的时候。只要按照题目的条件要求给图上加上点和边权，然后从[(1,1),K]开始跑一条最短路就好了。

       迪杰斯特拉是（(m+n)logn）的，这个题的话，点大概是110000个，边的话，每个点最多连出去k+4个点，理论上应该是，不会爆炸的。

---

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
using namespace std;
typedef long long ll;
const int maxn = 120005;
const int maxm=600005;
const int inf=(int)1e9;
const int N=105;
//走到点i用了j次机会的最短路
int cnt,n,K,A,B,C,mp[N][N],vis[maxn];
int k,now,w[maxm],dir[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
int to[maxm],nex[maxm],head[maxn];
ll dis[maxn];
struct  Node{
    int id;
    ll dist;
    Node(int id,ll dist):id(id),dist(dist){}
    bool operator < (const Node &a)const{
        return dist>a.dist;
    }
};

void add(int u,int v,int va){
    //printf("from = %d  to = %d  va = %d\n",u,v,va);
    to[cnt]=v; nex[cnt]=head[u];
    w[cnt]=va; head[u]=cnt++;
}
int gain(int x,int y,int oil){
    //printf("x = %d  y = %d oil = %d  p = %d\n",x,y,oil,(x-1)*n+y+(oil-1)*n*n);
    return (x-1)*n+y+oil*n*n;
}
int ck(int x,int y){
    return x>0&&x<=n&&y>0&&y<=n;
}
void dij(){
    for(int i=0;i<maxn;i++) {
        dis[i]=1e18; vis[i]=0;
    }
    priority_queue<Node> q;
    q.push(Node(gain(1,1,K),0));
    dis[gain(1,1,K)]=0;
    while(!q.empty()){
        Node u = q.top();  q.pop();
        if(vis[u.id]) continue;
        vis[u.id]=1;
        for(int i=head[u.id];~i;i=nex[i]){
            int v=to[i];
            if(vis[v]) continue;
            if(dis[v]-w[i]>dis[u.id]){
                dis[v]=(ll)dis[u.id]+w[i];
                q.push(Node(v,dis[v]));
            }
        }

    }
}
int main(){
    //freopen("trav4.in","r",stdin);
    memset(head,-1,sizeof(head));
    scanf("%d%d%d%d%d",&n,&K,&A,&B,&C);
    rep(i,1,n)
        rep(j,1,n){
            int x;
            scanf("%d",&x);
            mp[i][j]=x;
            rep(k,0,K-1){
                if(x==1) add(gain(i,j,k),gain(i,j,K),A);
                else add(gain(i,j,k),gain(i,j,K),A+C);
            }
        }

    rep(i,1,n){
        rep(j,1,n){
            rep(k,0,3){
                int dx=i+dir[k][0],dy=j+dir[k][1];
                if(!ck(dx,dy)) continue;
                if(mp[i][j]) add(gain(i,j,K),gain(dx,dy,K-1),(k<2?0:B));
                else{
                    rep(h,1,K){
                        add(gain(i,j,h),gain(dx,dy,h-1),(k<2?0:B));
                    }
                }
            }
        }
    }
    dij();
    ll ans=1e18;

    for(int i=0;i<=K;i++){
        //printf("i = %d dis = %lld\n",gain(n,n,i),dis[gain(n,n,i)]);
        ans=min(ans,dis[gain(n,n,i)]);
    }
    printf("%lld\n",ans);
    return 0;
}