原题链接
这个题感觉像是紧急救援那道题的复杂版,多加了几个条件,一样是最短路加上路径记录。由于数据范围很小,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;
}

更多推荐