洛谷P4009 汽车加油行驶问题

虽然写着网络流，但是似乎spfa跑最短路更好做一些。
需要注意的是如果某个地方有油站，那么是一定要加满油才能往后走的。
然后距离的记录以及边与平时的spfa有一些不同。
每个点如果有油，可以往4个方向转移。
别的就没有什么了。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<vector>
using namespace std;
struct node{
    int x,y,k;
};
queue<node>que;
int dis[105][105][15],fuel[105][105];//dis记录距离,x,y记录坐标而k记录油量
bool inque[105][105][15];
int main()
{
    int n,k,A,B,C,i,j;
    int move1[4][2]={1,0,-1,0,0,1,0,-1};
    cin>>n>>k>>A>>B>>C;que.push(node{1,1,k});memset(dis,0x3f,sizeof(dis));
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            cin>>fuel[i][j];
    dis[1][1][k]=0;//注意初始化
    while(!que.empty()){
        node node1=que.front();que.pop();
        inque[node1.x][node1.y][node1.k]=false;
        if(fuel[node1.x][node1.y]&&node1.k!=k){
            if(dis[node1.x][node1.y][k]>dis[node1.x][node1.y][node1.k]+A){
                dis[node1.x][node1.y][k]=dis[node1.x][node1.y][node1.k]+A;
                if(!inque[node1.x][node1.y][k]){
                    que.push(node{node1.x,node1.y,k});inque[node1.x][node1.y][k]=true;
                }
            }
            continue;//如果这个点有油站,那么一定必须先把油加满再去考虑下一步
        }
        else{//可以添加 油站
            if(dis[node1.x][node1.y][k]>dis[node1.x][node1.y][node1.k]+A+C){
                dis[node1.x][node1.y][k]=dis[node1.x][node1.y][node1.k]+A+C;
                if(!inque[node1.x][node1.y][k]){
                    que.push(node{node1.x,node1.y,k});inque[node1.x][node1.y][k]=true;
                }
            }
        }
        if(node1.k){//如果有油才能往后走
            for(i=0;i<4;i++){
                if(node1.x+move1[i][0]<1||node1.x+move1[i][0]>n||node1.y+move1[i][1]<0||node1.y+move1[i][1]>n)
                    continue;
                if(move1[i][0]<0||move1[i][1]<0){
                    if(dis[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]>dis[node1.x][node1.y][node1.k]+B){
                        dis[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]=dis[node1.x][node1.y][node1.k]+B;
                        if(!inque[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]){
                            inque[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]=true;
                            que.push(node{node1.x+move1[i][0],node1.y+move1[i][1],node1.k-1});
                        }
                    }
                }
                else{
                    if(dis[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]>dis[node1.x][node1.y][node1.k]){
                        dis[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]=dis[node1.x][node1.y][node1.k];
                        if(!inque[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]){
                            inque[node1.x+move1[i][0]][node1.y+move1[i][1]][node1.k-1]=true;
                            que.push(node{node1.x+move1[i][0],node1.y+move1[i][1],node1.k-1});
                        }
                    }
                }
            }
        }
    }
    int ans=(1<<31)-1;
    for(i=0;i<=k;i++)
        ans=min(ans,dis[n][n][i]);//,cout<<dis[n][n][i]<<' ';
    cout<<ans<<endl;

    return 0;
}