l3直捣黄龙

_Mirai

已于 2023-04-18 21:30:29 修改

阅读量46

点赞数 1

分类专栏：天梯赛文章标签： c++ 图论算法

于 2023-04-18 21:27:35 首次发布

本文链接：https://blog.csdn.net/qq_73051419/article/details/130231652

版权

天梯赛专栏收录该内容

10 篇文章 0 订阅

订阅专栏

本题为三元素最短路，路径为第一元素，到起始点的节点数为第二元素，过程中的敌军数量和为第三元素

本题以字符串作为节点编号，可以用map映射成数字编号节点便于操作，结果要求输出最终路径，所以需要一个记录每个节点的前驱数组

而结果路径输出是按照字符串输出，所以还需要一个反映射map便于根据数字编号节点输出字符串编号节点


#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
#define ll long long
#define Mirai ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
#define COUT cout << "###" << endl;
#define pb push_back
using namespace std;
typedef pair<int,int> PII;
const int N=220;
struct Edge
{
    int v,w,num;
};
vector<Edge> g[N];
map<string,int> mp;//映射map
map<int,string> ump;//反映射map
int a[N];//存储每个点的敌军数
int dist[N];//最短距离
bool vis[N];
int cnt[N];//最短路径条数
int d[N];//距离起点距离多少个城镇
int path[N];//前驱节点
int sum[N];
int n,m,idx;
string s,t;
void dij()
{
    dist[mp[s]]=0;
    cnt[mp[s]]=1;
    path[mp[s]]=mp[s];
    sum[mp[s]]=0;
    priority_queue<PII,vector<PII>,greater<PII>> q;
    q.push({dist[mp[s]],mp[s]});
    while(q.size())
    {
        int u=q.top().second;
        q.pop();
        if(vis[u])continue;
        vis[u]=true;
        for(auto [v,w,num]:g[u])
        {
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                d[v]=d[u]+1;
                cnt[v]=cnt[u];
                path[v]=u;
                sum[v]=sum[u]+num;
                q.push({dist[v],v});
            }
            else if(dist[v]==dist[u]+w)
            {
                if(d[v]<d[u]+1)
                {
                    d[v]=d[u]+1;
                    path[v]=u;
                    sum[v]=sum[u]+num;
                    q.push({dist[v],v});
                }
                else if(d[v]==d[u]+1)
                {
                    if(sum[v]<sum[u]+num)
                    {
                        sum[v]=sum[u]+num;
                        path[v]=u;
                        q.push({dist[v],v});
                    }
                }
                cnt[v]+=cnt[u];
            }
        }
    }
}
void Print(int u)
{
    if(path[u]!=u)
    {
        Print(path[u]);
        cout<<"->"<<ump[u];
        return ;
    }
    cout<<ump[u];
}
int main()
{
    cin>>n>>m>>s>>t;
    idx=0;
    mp[s]=++idx;//起点编号为1
    ump[idx]=s;
    for(int i=1;i<=n;i++)
    {
        dist[i]=INF;
    }
    for(int i=2;i<=n;i++)
    {
        string adr;
        int x;
        cin>>adr>>x;
        mp[adr]=++idx;
        ump[idx]=adr;
        a[idx]=x;
    }
    for(int i=0;i<m;i++)
    {
        string u,v;
        int w;
        cin>>u>>v>>w;
        g[mp[u]].push_back({mp[v],w,a[mp[v]]});
        g[mp[v]].push_back({mp[u],w,a[mp[u]]});
    }
    dij();
    Print(mp[t]);
    cout<<endl;
    cout<<cnt[mp[t]]<<" "<<dist[mp[t]]<<" "<<sum[mp[t]];
    return 0;
}