AOJ 2200 Mr. Rito Post Office 最短路径+动态规划+谨慎+思维

我写了好多注释,一看就能看懂,这个题目我想了6,7个小时,一开始忽略了船的位置和要把船安置的位置一致的情况,补上就对了。

cpp 复制代码
#include <iostream>
using namespace std;
int inf = 0x3f3f3f3f, num[1007], dp[1007][207], L[207][207], S[207][207], N, M, R;
void init()
{
    for (int i = 1; i <= N; i++)
    {
        for (int j = 1; j <= N; j++)
        {
            L[i][j] = inf;
            S[i][j] = inf;
        }
        L[i][i] = 0;
        S[i][i] = 0;
    }
}
void input()
{
    int from, to, cost;
    char op;
    for (int i = 1; i <= M; i++)
    {
        scanf("%d %d %d %c\n", &from, &to, &cost, &op);
        if (op == 'S')
        {
            if (cost < S[from][to])
            {
                S[from][to] = cost;
            }
            if (cost < S[to][from])
            {
                S[to][from] = cost;
            }
        }
        else if (op == 'L')
        {
            if (cost < L[from][to])
            {
                L[from][to] = cost;
            }
            if (cost < L[to][from])
            {
                L[to][from] = cost;
            }
        }
    }
    scanf("%d", &R);
    for (int i = 1; i <= R; i++)
    {
        scanf("%d", &num[i]);
    }
}
void floyd()
{
    for (int k = 1; k <= N; k++)
    {
        for (int i = 1; i <= N; i++)
        {
            for (int j = 1; j <= N; j++)
            {
                if (L[i][k] != inf && L[k][j] != inf)
                {
                    if (L[i][k] + L[k][j] < L[i][j])
                    {
                        L[i][j] = L[i][k] + L[k][j];
                    }
                }
                if (S[i][k] != inf && S[k][j] != inf)
                {
                    if (S[i][k] + S[k][j] < S[i][j])
                    {
                        S[i][j] = S[i][k] + S[k][j];
                    }
                }
            }
        }
    }
}
void handleNormalLine(int i, int j)
{
    // dp[i][j]是从num[1]到达num[i],并且到达num[i]时船在j的最小路径,当i大于1时,dp[i][j]一定与num[i-2]走到num[i-1]时停船的位置有关
    // 我们需要从num[i-1]走陆路到k,然后走水路到j,把船停在j,之后走陆路从j到num[i]
    dp[i][j] = inf;
    for (int k = 1; k <= N; k++)
    {
        if (k != j)
        {
            // 从num[i-1]到k的陆路不通,从k到j的水路不通,从j到num[i]的陆路不通,从1到num[i-1]并且停船到k实现不了,那么这种情况不用计算
            if (L[num[i - 1]][k] == inf || S[k][j] == inf || L[j][num[i]] == inf || dp[i - 1][k] == inf)
            {
                continue;
            }
            if (L[num[i - 1]][k] + S[k][j] + L[j][num[i]] + dp[i - 1][k] < dp[i][j])
            {
                dp[i][j] = L[num[i - 1]][k] + S[k][j] + L[j][num[i]] + dp[i - 1][k];
            }
        }
        else
        {
            // k和j相等时,就不需要走陆路到k,然后再走水路到j了,直接从num[i-1]走陆路到num[i]即可,因为j==k船已经在j了,不用管船
            if (L[num[i - 1]][num[i]] == inf || dp[i - 1][k] == inf)
            {
                continue;
            }
            if (L[num[i - 1]][num[i]] + dp[i - 1][k] < dp[i][j])
            {
                dp[i][j] = L[num[i - 1]][num[i]] + dp[i - 1][k];
            }
        }
    }
}
void handleFirstLine(int i, int j)
{
    // i=1时,船就在num[i],dp[i][j] i=1 代表开船到j,船放在j,然后陆路走回来num[i]
    // 走水路开船从num[i]到j,然后船停在j,之后从j走陆路回到num[i],如果num[i]到j的水路,或者j到num[i]的陆路不通,那么这个都无法实现
    if (S[num[i]][j] == inf || L[j][num[i]] == inf)
    {
        dp[i][j] = inf;
        return;
    }
    dp[i][j] = S[num[i]][j] + L[j][num[i]];
}
void doDp()
{
    // 我们用dp[i][j] 代表邮递员从 num[1]按照顺序一个个走到num[i],即达到邮递员在num[i]且船的位置在j的状态下,最小的消耗
    for (int i = 1; i <= R; i++)
    {
        for (int j = 1; j <= N; j++)
        {
            if (i == 1)
            {
                handleFirstLine(i, j);
            }
            else
            {
                handleNormalLine(i, j);
            }
        }
    }
}
int findAns()
{
    int ans = inf;
    for (int i = 1; i <= N; i++)
    {
        if (dp[R][i] < ans)
        {
            ans = dp[R][i];
        }
    }
    return ans;
}
int main()
{
    while (true)
    {
        scanf("%d%d", &N, &M);
        if (N == 0 && M == 0)
        {
            break;
        }
        init();
        input();
        floyd();
        doDp();
        printf("%d\n", findAns());
    }
    return 0;
}
相关推荐
月落归舟1 小时前
帮你从算法的角度来认识二叉树---(二)
算法·二叉树
清华都得不到的好学生2 小时前
数据结构->1.稀疏数组,2.数组队列(没有取模),3.环形队列
java·开发语言·数据结构
SilentSlot2 小时前
【数据结构】Hash
数据结构·算法·哈希算法
是娇娇公主~2 小时前
Lambda表达式详解
数据结构·c++
样例过了就是过了3 小时前
LeetCode热题100 柱状图中最大的矩形
数据结构·c++·算法·leetcode
wsoz3 小时前
Leetcode哈希-day1
算法·leetcode·哈希算法
阿Y加油吧3 小时前
LeetCode 二叉搜索树双神题通关!有序数组转平衡 BST + 验证 BST,小白递归一把梭
java·算法·leetcode
BlockChain8883 小时前
区块链的组件:从数据结构到去中心化共识
数据结构·去中心化·区块链
liuyao_xianhui4 小时前
优选算法_最小基因变化_bfs_C++
java·开发语言·数据结构·c++·算法·哈希算法·宽度优先
jimy14 小时前
C语言实现-----面向对象编程
c语言·数据结构