洛谷P4009 汽車加油行駛問題
阿新 • • 發佈:2019-02-15
雖然寫著網路流,但是似乎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;
}