1. 程式人生 > >洛谷P4009 汽車加油行駛問題

洛谷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; }