一个人的旅行-SPFA-map-构点

最新推荐文章于 2018-11-17 12:01:31 发布

原创最新推荐文章于 2018-11-17 12:01:31 发布 · 170 阅读

0 ·

CC 4.0 BY-SA版权

最短路专栏收录该内容

28 篇文章

订阅专栏

博客给出了HDU - 2066“一个人的旅行”问题的代码实现，通过强行加上初始点和结束点，使用SPFA算法求解。代码中定义了相关变量和数据结构，在主函数中处理输入，构建图，最后调用SPFA算法计算并输出结果。

W - 一个人的旅行
HDU - 2066
强行加上初始点结束点即可

#include<bits/stdc++.h>
using namespace std;
#define maxn 55555
#define inf 0x3f3f3f3f
vector<pair<int,int> >mp[maxn];
int t,s,d,dis[maxn],vis[maxn],u,v,w,cnt,ed;
map<int,int>g;
void spfa()
{
    for(int i=0; i<=ed; i++)
    {
        dis[i]=inf;
        vis[i]=0;
    }
    vis[0]=1;
    dis[0]=0;
    queue<int>q;
    q.push(0);
    while(!q.empty())
    {
        int top=q.front();
        vis[top]=0;
        q.pop();
        for(int i=0; i<mp[top].size(); i++)
        {
            if(dis[mp[top][i].first]>dis[top]+mp[top][i].second)
            {
                dis[mp[top][i].first]=dis[top]+mp[top][i].second;
                if(!vis[mp[top][i].first])
                {
                    q.push(mp[top][i].first);
                    vis[mp[top][i].first]=1;
                }
            }
        }
    }
}
int main()
{
    while(cin>>t>>s>>d)
    {
        g.clear();
        for(int i=0; i<=cnt; i++)
            mp[i].clear();
        cnt=0;
        while(t--)
        {
            cin>>u>>v>>w;
            if(g[u]==0)
                g[u]=++cnt;
            if(g[v]==0)
                g[v]=++cnt;
            mp[g[u]].push_back(make_pair(g[v],w));
            mp[g[v]].push_back(make_pair(g[u],w));
        }
        while(s--)
        {
            cin>>v;
            mp[0].push_back(make_pair(g[v],0));
        }
        ed=cnt+d+1;
        while(d--)
        {
            cin>>v;
            if(g[v]==0)
                g[v]=++cnt;
            mp[g[v]].push_back(make_pair(ed,0));
        }
        spfa();
        cout<<dis[ed]<<endl;
    }
    return 0;
}