天梯赛L3-011 直捣黄龙 c++

yaoye222

已于 2024-03-02 18:52:54 修改

阅读量157

点赞数 3

分类专栏：算法与数据结构文章标签： c++ 算法数据结构

于 2024-02-24 17:11:59 首次发布

本文链接：https://blog.csdn.net/yaoye222/article/details/136273735

版权

算法与数据结构专栏收录该内容

22 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Dijkstra算法求解最短路径问题，同时考虑路径经过城镇的数量和消灭敌军的数量，以及在更新路径时的路径计数策略。算法通过优先队列实现，并在给定的战争策略背景下进行演示。

摘要由CSDN通过智能技术生成

原题链接
这个题感觉像是紧急救援那道题的复杂版，多加了几个条件，一样是最短路加上路径记录。由于数据范围很小，N最大不超过200，所以可以直接把map当成超级数组用。
第一次交的时候测试点2没过，后来发现是最快进攻路径的条数这里出现了问题：

当点 j 的最短路径被 u 更新时，从起点到 j 的路径数等于 u 的路径数
当点 j 的最短路径等于 u 时，到 j 的路径数应当加上 u 的路径数

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,string> PIS;
#define X first
#define Y second
int n,k;
string Start,End;
// 敌军数量、到该点经过城镇数量、到该点消灭敌军数量、到该点距离、到该点路径数
map<string,int> enemy,cnt,num,dist,road;
map<string,vector<PIS>> v;
map<string,string> pre;
void dijkstra()
{
    priority_queue<PIS,vector<PIS>,greater<PIS>> q;
    q.push({0,Start});
    pre[Start]=Start,cnt[Start]=0,dist[Start]=0,num[Start]=0;
    road[Start]=1;
    while(!q.empty()) {
        string u=q.top().Y;
        int distance=q.top().X;
        q.pop();
        for(auto i:v[u]) {
            string j=i.Y;
            if(dist[j]>distance+i.X) {
                dist[j]=distance+i.X;
                pre[j]=u,road[j]=road[u];
                cnt[j]=cnt[u]+1,num[j]=num[u]+enemy[j];
                q.push({dist[j],j});
            } else if(dist[j]==distance+i.X) {
                if(cnt[j]<cnt[u]+1) {
                    cnt[j]=cnt[u]+1,num[j]=num[u]+enemy[j];
                    pre[j]=u;
                } else if(cnt[j]==cnt[u]+1) {
                    if(num[j]<num[u]+enemy[j]) {
                        num[j]=num[u]+enemy[j];
                        pre[j]=u;
                    }
                }
                road[j]+=road[u];
            }
        }
    }

    stack<string> ans;
    string p=End;
    while(p!=Start) {
        ans.push(p);
        p=pre[p];
    }
    cout<<Start;
    while(!ans.empty()) {
        string u=ans.top();ans.pop();
        cout<<"->"<<u;
    }
    cout<<'\n'<<road[End]<<' '<<dist[End]<<' '<<num[End]<<'\n';
}
int main()
{
    ios::sync_with_stdio(false),cin.tie(0);
    cin>>n>>k>>Start>>End;
    for(int i=1;i<n;i++) {
        string name;
        int number;
        cin>>name>>number;
        enemy[name]=number;
        dist[name]=0x3f3f3f3f;
    }
    while(k--) {
        string a,b;
        int cost;
        cin>>a>>b>>cost;
        v[a].push_back({cost,b});
        v[b].push_back({cost,a});
    }
    dijkstra();
    return 0;
}