P4009 汽车加油行驶问题

最新推荐文章于 2022-07-14 19:45:06 发布

转载最新推荐文章于 2022-07-14 19:45:06 发布 · 184 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/clno1/p/10710267.html

本文介绍了一种使用Dijkstra算法解决特定场景下汽车路径规划问题的方法。在一个N×N的网格中，汽车从左上角出发，目标是到达右下角，途中需考虑油库位置、行驶成本及加油费用等因素，算法旨在找到总费用最低的行驶路线。

题意：给定一个 N×N

设计好状态直接跑Dijkstra最短路即可。

#include<bits/stdc++.h>
using namespace std;
const int N=110;
const int INF=0x3f3f3f3f;
const int dx[]={-1,0,1,0};
const int dy[]={0,1,0,-1};
int n,k,A,B,C,mp[N][N];
struct node{
    int x,y,k;
    int d;
    node() {}
    node(int x,int y,int k,int d) : x(x),y(y),k(k),d(d) {}
    bool operator < (const node &rhs) const {
        return d>rhs.d;
    }
};

int d[N][N][15]; bool vis[N][N][15];
priority_queue<node> q; 
int Dijkstra() {
    memset(d,0x3f,sizeof(d));
    memset(vis,0,sizeof(vis));
    d[1][1][k]=0; q.push(node(1,1,k,0));
    int ret=INF;
    while (!q.empty()) {
        node u=q.top(); q.pop();
        if (u.x==n && u.y==n) ret=min(ret,u.d);
        if (vis[u.x][u.y][u.k]) continue;
        vis[u.x][u.y][u.k]=1;
        
        for (int i=0;i<4;i++) {  //走路 
            int nx=u.x+dx[i],ny=u.y+dy[i],nk=u.k-1;
            if (nx<1 || nx>n || ny<1 || ny>n || nk<0) continue;
            int tmp=d[u.x][u.y][u.k];
            if (nx<u.x || ny<u.y) tmp+=B;
            if (mp[nx][ny]==1) { tmp+=A; nk=k; }
            if (tmp<d[nx][ny][nk]) {
                d[nx][ny][nk]=tmp;
                q.push(node(nx,ny,nk,tmp)); 
            }
        }
        
        if (u.k==0 && mp[u.x][u.y]==0) {  //设立油库并加油 
            int tmp=d[u.x][u.y][u.k]+C+A;
            if (tmp<d[u.x][u.y][k]) {
                d[u.x][u.y][k]=tmp;
                q.push(node(u.x,u.y,k,tmp));
            }
        }
    }
    return ret;    
}

int main()
{
    cin>>n>>k>>A>>B>>C;
    for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) scanf("%d",&mp[i][j]);    
    
    cout<<Dijkstra()<<endl;
    return 0;
}