PAT A1030 Travel Plan (30 分) 图 Dijkstra

    题目大意:给出图的信息,每条边的边权有两个,一是distance, 二是 cost,求出从起点到目的地的最短路径,如果最短路径有多条,输出 cost 最小的一条。

    常规的dijkstra + dfs解法。

AC代码:

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>

using namespace std;

const int MAXN = 510;
const int INF = 0x7FFFFFFF;
int G[MAXN][MAXN] = {0};
int d[MAXN] = {0};
int cost[MAXN][MAXN] = {0};
bool visited[MAXN] = {false};
vector<int> pre[MAXN];

void dijkstra(int s, int N)
{
    d[s] = 0;
    for (int i = 0; i < N; ++i)
    {
        int u = -1, min = INF;
        for (int j = 0; j < N; ++j)
        {
            if(!visited[j] && d[j] < min)
            {
                u = j;
                min = d[j];
            }
        }
        if(u == -1) return;
        visited[u] = true;
        for (int k = 0; k < N; ++k)
        {
            if(!visited[k] && G[u][k] < INF && d[u] + G[u][k] < d[k])
            {
                d[k] = d[u] + G[u][k];
                pre[k].clear();
                pre[k].push_back(u);
            }
            else if(!visited[k] && G[u][k] < INF && d[u] + G[u][k] == d[k])
                pre[k].push_back(u);
        }
    }
}

vector<int> path, tmpPath;
int optDis = INF, optCost = INF;
void dfs(int v, int s)
{
    if(v == s)
    {
        tmpPath.push_back(v);
        int dis = 0, cos = 0;
        for (int i = tmpPath.size() - 1; i >= 1; --i)
        {
            dis += G[tmpPath[i]][tmpPath[i-1]];
            cos += cost[tmpPath[i]][tmpPath[i-1]];
        }
        if(cos < optCost)
        {
            optDis = dis;
            optCost = cos;
            path = tmpPath;
        }
        tmpPath.pop_back();
        return;
    }
    tmpPath.push_back(v);
    for (int i = 0; i < pre[v].size(); ++i)
    {
        dfs(pre[v][i], s);
    }
    tmpPath.pop_back();
}

int main()
{
    fill(G[0], G[0] + MAXN * MAXN, INF);
    fill(d, d + MAXN, INF);
    int N, M, S, D;
    cin >> N >> M >> S >> D;
    for (int i = 0; i < M; ++i)
    {
        int u, v, l, w;
        scanf("%d%d%d%d", &u, &v, &l, &w);
        G[u][v] = G[v][u] = l;
        cost[u][v] = cost[v][u] = w;
    }
    dijkstra(S, N);
    dfs(D, S);
    for (int i = path.size() - 1; i >= 0; --i)
    {
        printf("%d ", path[i]);
    }
    printf("%d ", optDis);
    printf("%d", optCost);
    return 0;
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值